[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;
    }
};