面试题 02.04. 分割链表(LeetCode)

简介: 面试题 02.04. 分割链表(LeetCode)

4989d0d4e72712af1c0575f8e6522fc4_6c7b42a7d9284623ae95ec6507ec7a1c.png我们上点强度,如果要保留各分区中节点的初始相对位置,应该怎么办?


此时想法就是,创建两个lesshead和morehead链表指针,把小于x的节点放在lesshead链表里,把大于等于x的节点放在morehead链表里,最后再将两个链表链接起来


但是这里只用指针的话,会遇到空链表的情况要分类讨论,比较麻烦,所以这里我们可以创建哨兵位,用带头链表会比较方便

7c719cf5f9fd5fdcaf36a0ab7cb83253_904389645259484bb0497c18934f859d.png


但如果你是这样写的话,可能会遇到以下报错,这是为什么呢?

84bb7b7b69e01e434b022e11f0c6b6af_b630dd6e75d7431db16e92474115634b.png


这是因为,你创建的哨兵位中next没有初始化,所以要将其置为NULL

537b608c03477a0c564ed96136bbf269_65de2d2802e044aa8652aa3d892912ab.png


在判断的时候,如果cur指向的值小于x,则链接到lesstail上,反之,则链接到moretail上。不过,此时要注意,链接完以后,应该及时将尾部节点next置为NULL,否则后续可能会出现环状链表

72cbb87fe94cc707efcd115dfffbcab8_52a991f7011e486580f06fcaffa291fb.png


链接完两个链表后,记得要把哨兵位的空间给释放掉

8b2fa673bc012f35fc1cff8a7f46aa30_6feba6dd6a064e94b2f3ea07bf47ec56.png


最终代码如下:


struct ListNode* partition(struct ListNode* head, int x)
{
    struct ListNode* lesshead , *lesstail, *morehead, *moretail;
    lesshead = lesstail = (struct ListNode*)malloc(sizeof(struct ListNode));
    lesshead->next = NULL;
    morehead = moretail = (struct ListNode*)malloc(sizeof(struct ListNode));
    morehead->next = NULL;
    struct ListNode* cur = head;
    while (cur)
    {
        if (cur->val < x)
        {
            lesstail->next = cur;
            lesstail = lesstail->next;
            cur = cur->next;
            lesstail->next = NULL;
        }
        else
        {
            moretail->next = cur;
            moretail = moretail->next;
            cur = cur->next;
            moretail->next = NULL;
        }
    }
    lesstail->next = morehead->next;
    free(morehead);
    head = lesshead->next;
    free(lesshead);
    return head;
}
相关文章
|
22天前
【力扣】-- 移除链表元素
【力扣】-- 移除链表元素
31 1
|
29天前
Leetcode第21题(合并两个有序链表)
这篇文章介绍了如何使用非递归和递归方法解决LeetCode第21题,即合并两个有序链表的问题。
46 0
Leetcode第21题(合并两个有序链表)
|
29天前
LeetCode第二十四题(两两交换链表中的节点)
这篇文章介绍了LeetCode第24题的解法,即如何通过使用三个指针(preNode, curNode, curNextNode)来两两交换链表中的节点,并提供了详细的代码实现。
15 0
LeetCode第二十四题(两两交换链表中的节点)
|
29天前
Leetcode第十九题(删除链表的倒数第N个节点)
LeetCode第19题要求删除链表的倒数第N个节点,可以通过快慢指针法在一次遍历中实现。
38 0
Leetcode第十九题(删除链表的倒数第N个节点)
|
29天前
|
索引
力扣(LeetCode)数据结构练习题(3)------链表
力扣(LeetCode)数据结构练习题(3)------链表
72 0
|
29天前
【LeetCode 10】142. 环形链表 II
【LeetCode 10】142. 环形链表 II
19 0
|
29天前
【LeetCode 09】19 删除链表的倒数第 N 个结点
【LeetCode 09】19 删除链表的倒数第 N 个结点
14 0
|
29天前
【LeetCode 08】206 反转链表
【LeetCode 08】206 反转链表
12 0
|
29天前
【LeetCode 06】203.移除链表元素
【LeetCode 06】203.移除链表元素
28 0
|
1月前
【数据结构】环形、相交、回文、分割、合并、反转链表
【数据结构】环形、相交、回文、分割、合并、反转链表
25 0