标签:des style blog class code c
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小,一个大于等于x。最后把它们连起来。注意把第二个链表最后一个元素置为NULL。不然可能产生死循环。
1 class Solution { 2 public: 3 ListNode *partition(ListNode *head, int x) { 4 ListNode n1(0), n2(0); 5 ListNode *p1 = &n1, *p2 = &n2; 6 7 while (head != NULL) { 8 if (head->val < x) { 9 p1->next = head; 10 p1 = p1->next; 11 } else { 12 p2->next = head; 13 p2 = p2->next; 14 } 15 head = head->next; 16 } 17 p2->next = NULL; 18 p1->next = n2.next; 19 return n1.next; 20 } 21 };
这里n1和n2不用new,省得最后还要delete.
Leetcode | Partition List,布布扣,bubuko.com
标签:des style blog class code c
原文地址:http://www.cnblogs.com/linyx/p/3734004.html