给定一个单链表和数值x,划分链表使得所有小于x的节点排在大于等于x的节点之前。
你应该保留两部分内链表节点原有的相对顺序。
您在真实的面试中是否遇到过这个题?
Yes
样例
给定链表 1->4->3->2->5->2->null,并且 x=3
返回 1->2->2->4->3->5->null
/**
* Definition of ListNode
* class ListNode {
* public:
* int val;
* ListNode *next;
* ListNode(int val) {
* this->val = val;
* this->next = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param head: The first node of linked list.
* @param x: an integer
* @return: a ListNode
*/
ListNode *partition(ListNode *head, int x) {
// write your code here
ListNode * h1=new ListNode();
ListNode * h2=new ListNode ();
ListNode * h1_cur=h1;
ListNode * h2_cur=h2;
ListNode* p=head;
if(p==NULL){
return head;
}
while(p!=NULL){
ListNode * temp=p;
p=p->next;
temp->next=NULL;
if(temp->val< x){
h1_cur->next=temp;
h1_cur=h1_cur->next;
}
else{
h2_cur->next=temp;
h2_cur=h2_cur->next;
}
}
if(h1->next==NULL){
return h2->next;
}
if(h2->next==NULL){
return h1->next;
}
h1_cur->next=h2->next;
return h1->next;
}
};