LeetCode: Partition List

AIvin24

于 2014-08-23 11:44:22 发布

阅读量475

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/AIvin24/article/details/38777247

LeetCode 专栏收录该内容

144 篇文章

订阅专栏

本文介绍了一种链表分隔算法的实现思路及代码。该算法通过新建两个空节点来分别记录小于x和大于等于x的子链，并最终将这两部分连接起来。文章详细解释了如何避免产生环的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：新建两个空节点（为了边界条件的方便），分别记录小于 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;
    }
};