本文讲解链表分割的有相对顺序和无相对顺序的版本【提供C++版本的代码】
一、【无相对顺序】分割链表
示例: 解题思路:
给一个头head和一个尾tail,<x值的头插,>=x值的尾插即可。毕竟不要求相对顺序。只要求<x的节点在前面,>=x的节点在后面即可。如果既要头插又要尾插的话就不建议创建一个头结点了,因为夹在中间,他还没用,会造成妨碍。
C++代码:
ListNode* partition(ListNode* phead, int x) {
ListNode* head = nullptr;
ListNode* tail = nullptr;
ListNode* cur = phead;
ListNode* next = nullptr;
while (cur) {
if (cur->val < x) {//<x则头插
if (head == nullptr) {
//head为空说明tail也为空,头插head即使为空也不用管
//但是tail后面还要用,所以必须更新,所以要加这一判断逻辑
head = tail = cur;
cur = cur->next;
}
else {//头插就没tail的事了
next = cur->next;
cur->next = head;
head = cur;
cur = head;
}
}
else {//>=x则尾插
if (head == nullptr) {
head = tail = cur;//尾插肯定也要更新
}
else {
tail->next = cur;
tail = cur;
}
cur = cur->next;
}
}
if (tail) {
tail->next = nullptr; // 尾结点指针置空
}
return head;
}
二、【有相对顺序】分割链表
示例:
解题思路:
本题难在不能改变相对顺序。
那么创建两个链表,一个链表尾插<x的值,一个链表尾插>=x的值,然后再把两个链表顺序链接即可。并且要给两个链表定义尾指针,lessTail和greaterTail,且要开辟一个头结点,这两个操作都是为方便尾插。因为没有头结点还要多加一个判断。
如果开辟头结点,还要多两个操作:
- 1、OJ题中默认都是没头结点的,如果要返回头指针,一定是要返回头结点的下一节点的地址,并且在两个链表链接过程中,一定是链接第二个链表的头结点的后一个节点。
- 2、要释放节点的内存,要返还给操作系统。【C或C++中】
小知识:一般上oj上是不会限制空间复杂度的,如果出现报错,内存限制:您的程序使用了超出限制的内存。这种问题可能是由死循环等的问题造成的。
C++代码:
class Solution {
public:
ListNode* partition(ListNode* head, int x) {
ListNode* lesshead, *lesstail, *greaterhead, *greatertail;
lesshead = lesstail = new ListNode();
greaterhead = greatertail = new ListNode();
ListNode* cur = head;//cur用来遍历链表
while (cur)
{
if (cur->val < x)
{
lesstail->next = cur;
lesstail = cur;
}
else
{
greatertail->next = cur;
greatertail = cur;
}
cur = cur->next;
}
lesstail->next = greaterhead->next;
greatertail->next = nullptr;
ListNode* newhead = lesshead->next;
delete lesshead;
delete greaterhead;
return newhead;
}
};
注:有相对顺序的代码提交到无相对顺序的也可以成功,因为本来就不要求顺序,个人认为有相对顺序的代码的处理逻辑更简单,因为有了头结点,判空逻辑就不用了!