标签:
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