【程序员面试金典】面试题 02.04. 分割链表

题目描述

描述：给你一个链表的头节点 head 和一个特定值 x ，请你对链表进行分隔，使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。

你不需要 保留 每个分区中各节点的初始相对位置。

输入：head = [1,4,3,2,5,2], x = 3
输出：[1,2,2,4,3,5]

输入：head = [2,1], x = 2
输出：[1,2]

1、链表中节点的数目在范围 [0, 200] 内；
2、-100 <= Node.val <= 100；
3、-200 <= x <= 200；

解题思路

思路1：如果这是数组，我们完全可以使用双指针，一头一尾，但是这个是单链表，这就意味着我们不知道尾指针，也无法从尾指针往前遍历，故需要另辟蹊径啦！我们可以遍历原始链表，将小于x的放在原链表，而将大于等于x的放在新链表，然后再将新链表拼接在原链表后面，这不就可以实现了嘛！

ListNode* partition(ListNode* head, int x) {
        // 原始链表
        ListNode* L=new ListNode(0);
        L->next=head;
        // 存储大于等于
        ListNode* G=new ListNode(0);
        G->next=NULL;
        // 遍历结点
        ListNode* p=L;
        ListNode* q=G;
        ListNode* temp;
        while(p&&p->next)
        {
            if(p->next->val>=x)
            {
                temp=p->next;
                p->next=p->next->next;
                q->next=temp;
                q=q->next;
            }
            else
                p=p->next;
        }
        //最后一个指针置空 否则计算机会认为你没有完全建立
        q->next=NULL;
        // p指向最后一个 p->next为空
        //将大于等于的头结点插入到最后
        p->next=G->next; 
        return L->next;
    }

总结：当你将两个链表拼接时，你需要将后一个链表的最后一个指针的下一个指针置为空，否则计算机以为你没有完全建立链表，于是报错heap-use-after-free on address。