首页 > 代码库 > leetcode--Partition List
leetcode--Partition List
Given a linked list and a value x, partition it such that all nodes less than x come before nodes greater than or equal to x.
You should preserve the original relative order of the nodes in each of the two partitions.
For example,
Given 1->4->3->2->5->2
and x = 3,
return 1->2->2->4->3->5
.
算法:找到第一个>=x的节点,记为p,p前一个节点记为s,向后遍历链表,如果节点的值小于x,就把该节点插入到s后面,同时s后移一位。
为了方便,使用了哨兵节点。
java:
/** * Definition for singly-linked list. * public class ListNode { * int val; * ListNode next; * ListNode(int x) { * val = x; * next = null; * } * } */ public class Solution { public ListNode partition(ListNode head, int x) { if(head==null||head.next==null) return head; ListNode h =new ListNode(-1); h.next=head; ListNode p=head; ListNode s=h; while(p!=null){ if(p.val>=x){ break; } s=s.next; p=p.next; } ListNode q=p; while(p!=null){ if(p.val<x){ ListNode tmp=p.next; q.next=tmp; p.next=s.next; s.next=p; s=s.next; } q=p; p=p.next; } return h.next; } }
c++:
/** * Definition for singly-linked list. * struct ListNode { * int val; * ListNode *next; * ListNode(int x) : val(x), next(NULL) {} * }; */ class Solution { public: ListNode *partition(ListNode *head, int x) { if(!head||!head->next) return head; ListNode * h =(ListNode *)malloc(sizeof(ListNode)); h->next=head; ListNode *p=head; ListNode *s=h; while(p){ if(p->val>=x){ break; } s=s->next; p=p->next; } ListNode *q=p; while(p){ if(p->val<x){ ListNode *tmp=p->next; q->next=tmp; p->next=s->next; s->next=p;
s=s->next;//后移一步 } q=p; p=p->next; } return h->next; } };
leetcode--Partition List
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。