我们上点强度,如果要保留各分区中节点的初始相对位置,应该怎么办?
此时想法就是,创建两个lesshead和morehead链表指针,把小于x的节点放在lesshead链表里,把大于等于x的节点放在morehead链表里,最后再将两个链表链接起来
但是这里只用指针的话,会遇到空链表的情况要分类讨论,比较麻烦,所以这里我们可以创建哨兵位,用带头链表会比较方便
但如果你是这样写的话,可能会遇到以下报错,这是为什么呢?
这是因为,你创建的哨兵位中next没有初始化,所以要将其置为NULL
在判断的时候,如果cur指向的值小于x,则链接到lesstail上,反之,则链接到moretail上。不过,此时要注意,链接完以后,应该及时将尾部节点next置为NULL,否则后续可能会出现环状链表
链接完两个链表后,记得要把哨兵位的空间给释放掉
最终代码如下:
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;
}