链表 2.4

编写代码，以给定值x为基准将链表分割成两部分，所有小于x的结点排在大于或者等于x的结点之前。

分析：使用两个链表，若当前结点小于x，则将其插入第一个链表的尾部；否则，将其插入第二个链表的尾部。最后，将两个链表拼接即可。

 1 //struct Node {
 2 //    Node(): val(0), next(0) {}
 3 //    Node( int value ): val(value), next(0) {}
 4 //    int val;
 5 //    Node *next;
 6 //};
 7
 8 Node* partition( Node *head, int x ) {
 9     Node guard1, guard2;
10     Node *end1 = &guard1, *end2 = &guard2;
11     while( head ) {
12         if( head->val < x ) {
13             end1->next = head;
14             end1 = head;
15         } else {
16             end2->next = head;
17             end2 = head;
18         }
19         head = head->next;
20     }
21     end2->next = 0;
22     end1->next = guard2.next;
23     return guard1.next;
24 }

运行结果

input:
target: 1
result:

input: 2
target: 1
result:2 

input: 13 1
target: 1
result:13 1 

input: 3 1 3
target: 2
result:1 3 3 

input: 1 2 3
target: 2
result:1 2 3 

input: 14 54 96 23 45 12 45
target: 23
result:14 12 54 96 23 45 45 

input: 14 54 96 23 45 12 45
target: 2
result:14 54 96 23 45 12 45 

input: 14 54 96 23 45 12 45
target: 24
result:14 23 12 54 96 45 45