给定一个单链表和数值x,划分链表使得所有小于x的节点排在大于等于x的节点之前。
你应该保留两部分内链表节点原有的相对顺序。
样例
给定链表 1->4->3->2->5->2->null,并且 x=3
返回 1->2->2->4->3->5->null
思路:
建立两个链表存放目标链表的结点,一个放数值小于给定的值的结点,一个放数值大于等于给定值的结点。在head节点向下移动并且插入到新建的链表中时,同时也不会改变原来节点的顺序。当head移动到最后,目标链表分成了两个链表,将数值大的链表整体插入到数值小的链表中即可。
代码:
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 *small=new ListNode(0);
ListNode *large=new ListNode(0);
ListNode *lastsmall=small;
ListNode *lastlarge=large;
ListNode *m=head;
while(m)
{ if(m->val<x)
{ lastsmall->next=m;
lastsmall=lastsmall->next;
}
else
{ lastlarge->next=m;
lastlarge=lastlarge->next;
}
m=m->next;
lastsmall->next=lastlarge->next=NULL;
}
lastsmall->next=large->next;
return small->next;
}
};
感想: