Reverse Linked List II
Reverse a linked list from position m to n. Do it in one-pass.
Note: 1 ≤ m ≤ n ≤ length of list.
Example:
Input: 1->2->3->4->5->NULL, m = 2, n = 4
Output: 1->4->3->2->5->NULL
分析:
首先需要找到待反转的子序列,然后对子序列进行n-m次反转。
反转过程如下(prev cur):
反转前:1->2->3->4->5->NULL
第1次:1->3->2->4->5->NULL
第2次:1->4->3->2->5->NULL
从过程中我们可以看出第1次置换了2和3的位置,此时2和3在序列中已经反转,那么在第2次只需要将4放到子序列的头部即可,同理,如果n = 5的话,此时2,3,4在序列中已经反转,下一次反转只需要将5放到子序列的头部即可。
简单来讲,就是在每次反转中cur之前的序列已经反转,然后将cur置于子序列的头部来完成下一次反转。
// Reverse Linked List II
// 迭代版,时间复杂度O(n),空间复杂度O(1)
class Solution {
public:
ListNode *reverseBetween(ListNode *head, int m, int n) {
ListNode dummy(-1);
dummy.next = head;
ListNode *prev = &dummy;
for (int i = 0; i < m-1; ++i)
prev = prev->next;
ListNode* const head2 = prev;
prev = head2->next;
ListNode *cur = prev->next;
for (int i = m; i < n; ++i) {
prev->next = cur->next;
cur->next = head2->next;
head2->next = cur; // 头插法
cur = prev->next;
}
return dummy.next;
}
};