LeetCode 92.Reverse Linked List II 反轉鏈表 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添加鏈接描述

思路

先找到第一個反轉的節點，然後將該節點的前面一個節點記爲頭節點，然後頭插法對後面的每個節點進行插入到頭節點之後。

eg: 1->2->3->4->5 ==> 1->3->2->4->5=> 1->4->3->2->5

c++

/**
 * 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) {
        ListNode dummy(-1);
        // 找到第一個需要反轉的節點
        dummy.next = head;
        ListNode *preNode = &dummy;
        for(int i = 0; i < m-1; i++){
            preNode = preNode->next;
        }
        ListNode *head2 = preNode;
        // 使用頭插法進行插入
        preNode = head2->next;
        ListNode *currNode = preNode->next;
        for(int i = m; i < n; i++){
            preNode->next = currNode->next;
            currNode->next = head2->next;
            head2->next = currNode;
            currNode = preNode->next;
        }
        return dummy.next;
    }
};