反转链表Ⅱ

泡椒香菜‍

于 2024-08-24 14:00:43 发布

阅读量530

点赞数 19

分类专栏： LeetCode 文章标签：链表数据结构 leetcode

本文链接：https://blog.csdn.net/weixin_52739099/article/details/141498813

版权

LeetCode 专栏收录该内容

19 篇文章 0 订阅

订阅专栏

反转链表Ⅱ

题目描述

Category	Difficulty	Likes	Dislikes
algorithms	Medium (55.89%)	1831	-

Tags

linked-list
Companies

Unknown

给你单链表的头指针 head 和两个整数 left 和 right ，其中 left <= right 。请你反转从位置 left 到位置 right 的链表节点，返回 反转后的链表。

示例 1：

输入：head = [1,2,3,4,5], left = 2, right = 4
输出：[1,4,3,2,5]

示例 2：

输入：head = [5], left = 1, right = 1
输出：[5]

提示：

链表中节点数目为 n
1 <= n <= 500
500 <= Node.val <= 500
1 <= left <= right <= n

进阶： 你可以使用一趟扫描完成反转吗？

求解

前言

链表的操作问题，一般而言面试（机试）的时候不允许我们修改节点的值，而只能修改节点的指向操作。

思路通常都不难，写对链表问题的技巧是：一定要先想清楚思路，并且必要的时候在草稿纸上画图，理清「穿针引线」的先后步骤，然后再编码。

方法一：穿针引线

思路

我们以下图中黄色区域的链表反转为例。

使用「反转链表」的解法，反转 left 到 right 部分以后，再拼接起来。我们还需要记录 left 的前一个节点，和 right 的后一个节点。如图所示：

算法步骤：

第 1 步：先将待反转的区域反转；
第 2 步：把 pre 的 next 指针指向反转以后的链表头节点，把反转以后的链表的尾节点的 next 指针指向 succ。

说明：编码细节我们不在题解中介绍了，请见下方代码。思路想明白以后，编码不是一件很难的事情。这里要提醒大家的是，链接什么时候切断，什么时候补上去，先后顺序一定要想清楚，如果想不清楚，可以在纸上模拟，让思路清晰。

代码

class Solution
{
public:
    ListNode *reverseBetween(ListNode *head, int left, int right)
    {
        ListNode *dummy = new ListNode(0, head);
        ListNode *left_p = dummy, *right_p = nullptr;

        for (int i = 0; i < left - 1; i++)
        {
            left_p = left_p->next;
        }
        right_p = left_p;
        for (int i = 0; i < right - left + 1; i++)
        {
            right_p = right_p->next;
        }
        ListNode *tmp = right_p->next;
        right_p->next = nullptr;
        ListNode *reverseSubHead = reverseList(left_p->next);

        left_p->next->next = tmp;
        left_p->next = reverseSubHead;

        ListNode *ans = dummy->next;
        delete dummy;
        return ans;
    }

    ListNode *reverseList(ListNode *head)
    {
        ListNode *pre = nullptr, *cur = head;
        while (cur)
        {
            ListNode *temp = cur->next;
            cur->next = pre;
            pre = cur;
            cur = temp;
        }
        return pre;
    }
};

复杂度分析

时间复杂度：O(N)，其中 N 是链表总节点数。最坏情况下，需要遍历整个链表。
空间复杂度：O(1)。只使用到常数个变量。

方法二：一次遍历「穿针引线」反转链表（头插法）

方法一的缺点是：如果 left 和 right 的区域很大，恰好是链表的头节点和尾节点时，找到 left 和 right 需要遍历一次，反转它们之间的链表还需要遍历一次，虽然总的时间复杂度为 O(N)，但遍历了链表 2 次，可不可以只遍历一次呢？答案是可以的。我们依然画图进行说明。

我们依然以方法一的示例为例进行说明。

整体思想是：在需要反转的区间里，每遍历到一个节点，让这个新节点来到反转部分的起始位置。下面的图展示了整个流程。

下面我们具体解释如何实现。使用三个指针变量 pre、curr、next 来记录反转的过程中需要的变量，它们的意义如下：

curr：指向待反转区域的第一个节点 left；
next：永远指向 curr 的下一个节点，循环过程中，curr 变化以后 next 会变化；
pre：永远指向待反转区域的第一个节点 left 的前一个节点，在循环过程中不变。
第 1 步，我们使用 ①、②、③ 标注「穿针引线」的步骤。

操作步骤：

先将 curr 的下一个节点记录为 next；
执行操作 ①：把 curr 的下一个节点指向 next 的下一个节点；
执行操作 ②：把 next 的下一个节点指向 pre 的下一个节点；
执行操作 ③：把 pre 的下一个节点指向 next。
第 1 步完成以后「拉直」的效果如下：

第 2 步，同理。同样需要注意「穿针引线」操作的先后顺序。

第 2 步完成以后「拉直」的效果如下：

第 3 步，同理。

第 3 步完成以后「拉直」的效果如下：

代码

class Solution {
public:
    ListNode *reverseBetween(ListNode *head, int left, int right) {
        // 设置 dummyNode 是这一类问题的一般做法
        ListNode *dummyNode = new ListNode(-1);
        dummyNode->next = head;
        ListNode *pre = dummyNode;
        for (int i = 0; i < left - 1; i++) {
            pre = pre->next;
        }
        ListNode *cur = pre->next;
        ListNode *next;
        for (int i = 0; i < right - left; i++) {
            next = cur->next;
            cur->next = next->next;
            next->next = pre->next;
            pre->next = next;
        }
        return dummyNode->next;
    }
};

复杂度分析：

时间复杂度：O(N)，其中 N 是链表总节点数。最多只遍历了链表一次，就完成了反转。
空间复杂度：O(1)。只使用到常数个变量。

方法三：栈

思路

与方法一相同的是，我们需要遍历一遍链表，记录下 left 和 right 节点的位置，并将其中的节点加入栈中。利用栈先进后出的数据结构达到指定区域节点的反转

算法

将 left_p 和 right_p 移动到 left 的前一个节点，如图
移动 right_p 到 right 节点的下一个节点，同时将移动中的节点加入栈中，如图
将 left_p 连接栈中第一个节点，然后重复连接后面的节点直到栈空
将栈的最后一个节点连接 right_p，如图
注意，连接栈中节点时需要将它们的 next 置为空，以避免出现环形链表的情况。

代码

class Solution
{
public:
    ListNode *reverseBetween(ListNode *head, int left, int right)
    {
        ListNode *dummy = new ListNode(0, head);
        ListNode *left_p = dummy, *right_p = nullptr;
        std::stack<ListNode *> stack;

        for (int i = 0; i < left - 1; i++)
        {
            left_p = left_p->next;
        }
        right_p = left_p;
        for (int i = 0; i < right - left + 1; i++)
        {
            right_p = right_p->next;
            stack.push(right_p);
        }
        right_p = right_p->next;

        while (!stack.empty())
        {
            left_p->next = stack.top();
            left_p->next->next = nullptr;
            stack.pop();
            left_p = left_p->next;
        }
        left_p->next = right_p;

        ListNode *ans = dummy->next;
        delete dummy;
        return ans;
    }
};