Leetcode Problem86

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.

猜你喜欢

转载自blog.csdn.net/vandance/article/details/82355124