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的值放置在x的前面,大于或者等于x的值放置在x的链表的后面。其实非常简单,我们定义两个新指针,从头遍历链表,小与x的节点挂在第一个指针后面,大于或者等于x的节点挂在第二个指针后面,最后将两个指针组成的短链表联合起来即可。
代码一:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
if(head==NULL) return NULL;
if(head->next==NULL) return head;
ListNode* p=head;
ListNode* s1=new ListNode(0);
ListNode* s2=new ListNode(0);
ListNode* ss1=s1;
ListNode* ss2=s2;
while(p)
{
if(p->val<x)
{
s1->next=p;
s1=s1->next;
}
else
{
s2->next=p;
s2=s2->next;
}
p=p->next;
}
if(ss2->next)
{
ss2=ss2->next;
s2->next=NULL;
}
else
ss2=NULL;
if(ss1->next)
{
s1->next=ss2;
ss1=ss1->next;
return ss1;
}
else
return ss2;
}
};代码一,简单直观,但是略有繁琐。下面我们稍作改动:
代码二:
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
if(head==NULL) return NULL;
if(head->next==NULL) return head;
ListNode* p=head;
ListNode* s1=new ListNode(0);
ListNode* s2=new ListNode(0);
ListNode* ss1=s1;
ListNode* ss2=s2;
while(p)
{
if(p->val<x)
{
s1=s1->next=p;
}
else
{
s2=s2->next=p;
}
p=p->next;
}
s2->next=NULL;
s1->next=ss2->next;
return ss1->next;
}
};注意:
代码二中s1=s1->next=p;指的是s1->next=p,之后再s1=p。也就是说s1节点指针跟p的属性相同。这是a=b=c形式,从右向左依次赋值,最后等式中的变量都和最右边的变量值相同;(a=b)=c 形式就不同了,虽然这种形式很少出现,指的是现将b的值付给a,a=b,之后再将c的值赋值给a,a=c;
相当于:
a=b;
a=c;
完成之后,b的值没有改变,c的值也没有改变,只是a=c而已。
版权声明:本文为博主原创文章,未经博主允许不得转载。
原文:http://blog.csdn.net/sinat_24520925/article/details/47176881