【技巧】LeetCode 86. Partition List

最新推荐文章于 2024-05-02 20:17:24 发布

Allenlzcoder

最新推荐文章于 2024-05-02 20:17:24 发布

阅读量464

点赞数

分类专栏： LeetCode练习题文章标签： LeetCode Partition List

本文链接：https://blog.youkuaiyun.com/Allenlzcoder/article/details/80347958

版权

LeetCode练习题专栏收录该内容

168 篇文章

订阅专栏

本文提供两种解决LeetCode 86题（分区列表）的方法。第一种方法使用额外空间创建两个新链表，再进行拼接；第二种方法通过调整链表指针实现，空间复杂度优化到O(1)。

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

LeetCode 86. Partition List

Solution1：
我的答案，时间复杂度 $O(n)$ ，空间复杂度 $O(n)$

/**
 * 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) {
        struct ListNode* smallHead = new ListNode(-1), *bigHead = new ListNode(-1), *cur = head;
        struct ListNode* smalltemp = smallHead, *bigtemp = bigHead;
        while (cur) {
            if (cur->val < x) {
                smalltemp->next = new ListNode(cur->val);
                smalltemp = smalltemp->next;
            }
            else {
                bigtemp->next = new ListNode(cur->val);
                bigtemp = bigtemp->next;
            }
            cur = cur->next;
        }
        smalltemp->next = bigHead->next;
        return smallHead->next;
    }
};

Solution2：
参考网址：http://www.cnblogs.com/grandyang/p/4321292.html
值得借鉴的技巧！！！
加了一个头结点，巧妙地避免了复制结点的操作，时间复杂度没变，但空间复杂度降为 $O(1)$

class Solution {
public:
    ListNode *partition(ListNode *head, int x) {
        if (!head) return head;
        ListNode *dummy = new ListNode(-1);
        ListNode *newDummy = new ListNode(-1);
        dummy->next = head;
        ListNode *cur = dummy, *p = newDummy;
        while (cur->next) {
            if (cur->next->val < x) {
                p->next = cur->next;
                p = p->next;
                cur->next = cur->next->next;
                p->next = NULL;
            } else {
                cur = cur->next;
            }
        }
        p->next = dummy->next;
        return newDummy->next;
    }
};