LeetCode: Partition List

最新推荐文章于 2018-10-12 09:53:13 发布

AIvin24

最新推荐文章于 2018-10-12 09:53:13 发布

阅读量452

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/AIvin24/article/details/38777247

版权

LeetCode 专栏收录该内容

144 篇文章 0 订阅

订阅专栏

思路：新建两个空节点（为了边界条件的方便），分别记录小于 x 的子链、大于等于 x 的子链，最后连接起来，注意应该断掉的节点，否则会产生环。

code:

class Solution {
public:
    ListNode *partition(ListNode *head, int x) {
        ListNode *p,*q, *t = head, *lessHead, *greaterHead;
        p = new ListNode(0);
        q = new ListNode(0);
        lessHead = p;
        greaterHead = q;
        while(t != NULL){
            if(t->val < x){
                p->next = t;
                p = p->next;
            }
            else{
                q->next = t;
                q = q->next;
            }
            t = t->next;
        }
        if(greaterHead->next == NULL || lessHead == NULL) 
            return head;
        p->next = greaterHead->next;
        q->next = NULL;
        head = lessHead->next;
        delete lessHead;
        delete greaterHead;
        return head;
    }
};