Partition List
问题描述:给定一个链表和一个数x,对其进行分区,使得小于x的所有节点都在大于或等于x的节点之前。
问题解决:这道题耗费了我一个半小时的时间。(好惨!!!)原来的思路是将小于x的节点组成一个链表,最后再拼接上剩余的节点,但是居然x可以不存在链表中,然后就一直过不了。后来看了网上别人的解法才知道怎么解决。首先设置一个辅助节点放在头节点前面(这一步很关键),然后找到第一个不小于x的节点(记为node),在这个节点之前的节点肯定是小于x的,不用移动,接下来沿着这个节点往下接着比较,如果小于x则将该节点放到node的前面,否则接着往下找。
class Solution {
public:
ListNode *partition(ListNode *head, int x) {
ListNode *add=new ListNode(-1);
add->next=head;
ListNode *pre=add,*cur;
//找到第一个不小于x的节点的前一个节点
while(pre->next!=NULL&&pre->next->val<x)
pre=pre->next;
cur=pre;
while(cur->next)
{
if(cur->next->val<x)
{
ListNode *tmp=cur->next;
cur->next=tmp->next;
tmp->next=pre->next;
pre->next=tmp;
pre=pre->next;
}
else
{
cur=cur->next;
}
}
return add->next;
}
};
beats 100.00 % of cpp submissions.