LeetCode 92. 反转链表 II

题目描述:

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

说明:
1 ≤ m ≤ n ≤ 链表长度。

示例:

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

思路:首先遍历,遍历的时候记录遍历过的节点数目,小于m的时候,继续扫描,记录反转链表前的最后一个节点。大于m小于n的时候,反转,大于n的时候,只需把后面的链表接在反转的那部分链表的链表尾部就可以了,然后再把开始反转的节点之前的节点的连接到反转链表的头部就可以了。

代码如下:

/**
 * 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 == nullptr || head->next == nullptr) return head;       
        ListNode *preNode = new ListNode(-1);
        preNode->next = head;
        ListNode *fStart = nullptr, *subStart = nullptr, *subEnd = nullptr, *pre = preNode, *p = head, *temp = nullptr;
        int s = 1;
        for(; p != nullptr && s <= m; ++s){
            if(s == m){
                fStart = pre;
                subEnd = p;
            }
            pre = p;
            p = p->next;
        }
        
        
        for(; p != nullptr && s <= n; ++s){
            if(s == n) subStart = p;
            temp = p;
            p = temp->next;
            temp->next = pre;
            pre = temp;  
        }
        
        if(fStart != nullptr && subStart != nullptr)//说明m大于链表的长度,没有交换
            fStart->next = subStart;
        subEnd->next = p;
        
        head = preNode->next;
        delete preNode;
        return head;
    }
};

猜你喜欢

转载自blog.csdn.net/sinat_15723179/article/details/81478655