反转从位置 m 到 n 的链表。请使用一趟扫描完成反转。
说明:
1 ≤ m ≤ n ≤ 链表长度。
示例:
输入: 1->2->3->4->5->NULL, m = 2, n = 4
输出: 1->4->3->2->5->NULL
关键问题在于寻找几个关键节点。
class Solution {
public:
ListNode *reverseBetween(ListNode *head, int m, int n) {
if (!head || m>n) return NULL;
ListNode* pNode = head;
ListNode* pre_head = NULL;
for (int i = m; i > 1; i--)
{
pre_head = pNode; //记录需要旋转部分头结点的前一个节点
pNode = pNode->next;//保存需要旋转部分的头结点
}
ListNode* cur = pNode;//需要旋转部分的头结点
ListNode* newhead = NULL;
ListNode* tail = NULL;//旋转区间的后一个节点
for (int i = m; i <= n;i++)
{
tail = cur->next; //更新旋转区间的后一个节点
ListNode* next = cur->next;
cur->next = newhead;
newhead = cur;
cur = next;
}
pNode->next = tail;
if (pre_head)//处理m=1的情况
{
pre_head->next = newhead;
return head;
}
else
return newhead;
}
};