题目描述
描述:给你一个链表的头节点 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。