首页 > 代码库 > 链表 2.4

链表 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: 1result:input: 2 target: 1result:2 input: 13 1 target: 1result:13 1 input: 3 1 3 target: 2result:1 3 3 input: 1 2 3 target: 2result:1 2 3 input: 14 54 96 23 45 12 45 target: 23result:14 12 54 96 23 45 45 input: 14 54 96 23 45 12 45 target: 2result:14 54 96 23 45 12 45 input: 14 54 96 23 45 12 45 target: 24result:14 23 12 54 96 45 45 

 

链表 2.4