Reorder List
描述
Given a singly linked list 0 \rightarrow L_1 \rightarrow \cdots \rightarrow L{n-1} \rightarrow Ln ,reorder it to: {n-1} \rightarrow L2 \rightarrow L{n-2} \rightarrow \cdots
You must do this in-place without altering the nodes' values.
For example,Given {1,2,3,4}
, reorder it to {1,4,2,3}
.
分析
题目规定要in-place,也就是说只能使用O(1)
的空间。
可以找到中间节点,断开,把后半截单链表reverse一下,再合并两个单链表。
代码
// Reorder List
// 时间复杂度O(n),空间复杂度O(1)
class Solution {
public:
void reorderList(ListNode *head) {
if (head == nullptr || head->next == nullptr) return;
ListNode *slow = head, *fast = head, *prev = nullptr;
while (fast && fast->next) {
prev = slow;
slow = slow->next;
fast = fast->next->next;
}
prev->next = nullptr; // cut at middle
slow = reverse(slow);
// merge two lists
ListNode *curr = head;
while (curr->next) {
ListNode *tmp = curr->next;
curr->next = slow;
slow = slow->next;
curr->next->next = tmp;
curr = tmp;
}
curr->next = slow;
}
ListNode* reverse(ListNode *head) {
if (head == nullptr || head->next == nullptr) return head;
ListNode *prev = head;
for (ListNode *curr = head->next, *next = curr->next; curr;
prev = curr, curr = next, next = next ? next->next : nullptr) {
curr->next = prev;
}
head->next = nullptr;
return prev;
}
};