92. 反转链表 II

反转从位置 m 到 n 的链表。请使用一趟扫描完成反转。

 

说明:

1 ≤ m ≤ n ≤ 链表长度。

 

示例:

输入: 1->2->3->4->5->NULL, m = 2, n = 4

输出: 1->4->3->2->5->NULL

 

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/reverse-linked-list-ii

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

寻找关键的节点

 

 

找到开始逆置的节点，记录该节点的前驱和该节点

 

从 head 开始，逆置 change_len = n-m+1 个节点

 

将 pre_head 与 new_head连接，modify_list_tail 与 head 连接

 

 

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};

class Solution {
public:
    ListNode* reverseBetween(ListNode* head, int m, int n) {
        int change_len = n - m + 1;
        ListNode *pre_head = NULL;//需要逆置的头节点的前驱
        ListNode *result = head;
        
        while (head && --m) {//m-1 次
            pre_head = head;//记录当前头节点
            head = head->next;
        }//head >>需要逆置的头节点
        ListNode *modify_list_tail = head;//记录初始逆置的头节点，也就是逆置后的尾部
        ListNode *new_head = NULL;//逆置后的头节点
        while (head && change_len) {
            ListNode *next = head->next;
            head->next = new_head;
            new_head = head;
            head = next;//最后一次 head 指向逆置段的后一个节点
            --change_len;//只逆置 change_len 个节点
        }
        modify_list_tail->next = head;//逆置段的尾部指向下一个节点
        if (pre_head) {
            pre_head->next = new_head;//逆置段的前一个节点指向
        } else {
            // pre_head = new_head; ❌
//             特殊情况： m=1 从第一个节点开始逆置，result存放的就是头节点，所以 result 需要变成逆置后的头节点
             result = new_head;
        }
        return result;
    }
};

 

测试

int main(int argc, const char * argv[]) {
    ListNode a(1);
    ListNode b(2);
    ListNode c(3);
    ListNode d(4);
    ListNode e(5);
    a.next = &b;
    b.next = &c;
    c.next = &d;
    d.next = &e;
    Solution solve;
    ListNode *head = solve.reverseBetween(&a, 2, 4);
    while (head) {
        cout <<head->val<<endl;
        head = head->next;
    }
    
    return 0;
}