[LeetCode系列]翻转链表问题II

给定一个链表和两个整数m, n, 翻转链表第m个节点到第n个节点(从1开始计数).

如, 给定链表: 1->2->3->4->5->NULL, 以及 m = 2, n = 4.

返回 1->4->3->2->5->NULL.

假定m和n满足约束条件: 1 ≤ m ≤ n ≤ 链表长度.

注意: 不能使用额外空间, 且只能遍历链表一次.

算法思路:

翻转的过程可以分解成3步:

把相邻的节点的指向关系倒置; 即 1->2<-3<-4 5->NULL

把第m-1个节点(1)指向第n个节点(4); 即 1->4->3->2 5->NULL

把第m个节点(2, 需要缓存)指向第n+1个节点(5). 即 1->4->3->2->5->NULL

/**
* Definition for singly-linked list.
* struct ListNode {
*     int val;
*     ListNode *next;
*     ListNode(int x) : val(x), next(NULL) {}
* };
*/
class Solution {
public:
ListNode *reverseBetween(ListNode *head, int m, int n) {
if(!head || m == n)
return head;
ListNode *p = head;
int count = 1;
while(count < m-1)
{
p = p->next;
count++;
}
ListNode *p1;
if(m == 1)
p1= p;
else
{
p1 = p->next;
count++;
}
ListNode *last = p1;
ListNode *p2 = p1->next;
ListNode *tmp = NULL;
while(p2 && count < n)
{
tmp = p2->next;
p2->next = p1;
p1 = p2;
p2 = tmp;
count++;
}
last->next = tmp;
if(m == 1)
head = p1;
else
p->next = p1;
return head;
}
};