牛客刷题-面试笔刷TOP101-7 链表内指定区间反转

链表内指定区间反转

将一个节点数为 size 链表 m 位置到 n 位置之间的区间反转，要求时间复杂度 O(n)，空间复杂度 O(1)。
例如：
给出的链表为 1→2→3→4→5→NULL*, m=2,n=4*，

返回 1→4→3→2→5→NULL

数据范围： 链表长度 0<size≤1000，0<m≤n≤size，链表中每个节点的值满足 ∣val∣≤1000

要求：时间复杂度 O(n)，空间复杂度 O(n)

进阶：时间复杂度 O(n)，空间复杂度 O(1)

示例1

输入：

{1,2,3,4,5},2,4

返回值：

{1,4,3,2,5}

示例2

输入：

{5},1,1

返回值：

{5}

解法1：使用栈

    public ListNode reverseBetween (ListNode head, int m, int n) {
        // write code here
        if (head == null) return null;
        Stack<int> stack = new Stack<int>();
        ListNode node = head;
        int index = 1;
        //遍历一遍，将m到n之间的元素存储到栈中
        while (node != null) {
            if (index >= m && index <= n) {
                stack.Push(node.val);
            }
            index++;
            node = node.next;
        }
        ListNode temp = head;
        index = 1;
        while (temp != null) {
            if (index >= m && index <= n) {
                temp.val = stack.Pop();
            }
            index++;
            temp = temp.next;
        }
        return head;
    }

解法2：直接反转

//第一种写法
public ListNode reverseBetween (ListNode head, int m, int n) {
        // write code here
        // 如果链表为空或 m 和 n 相等，则无需反转
        if (head == null || m == n) return head;

        // 初始化一个虚拟节点
        ListNode dummy = new ListNode(0);
        dummy.next = head;

        // 初始化指针
        ListNode leftNode = dummy;  // m 位置之前的节点
        ListNode current = head;

        // 将 leftNode 移动到 m 的前一个节点
        for (int i = 1; i < m; i++) {
            leftNode = current;
            current = current.next;
        }

        // 用来管理反转的指针
        ListNode reverseTail = current;
        ListNode prev = null;

        // 反转 m 到 n 的子链表
        for (int i = 0; i <= n - m; i++) {
            ListNode next = current.next;
            current.next = prev;
            prev = current;
            current = next;
        }

        // 将反转后的子链表与链表的其他部分重新连接
        leftNode.next = prev;
        reverseTail.next = current;

        // 返回新的链表头
        return dummy.next;
    }

//第二种写法
public ListNode reverseBetween (ListNode head, int m, int n) {
        // write code here
        //存储 m 的上一个结点
        ListNode leftNode = null;
        //存储 n 的下一个结点
        ListNode rightNode = null;
        //存储 反转后 的头结点
        ListNode reverseHead = null;
        //存储 反转后 的尾结点
        ListNode reverseTail = null;
        ListNode cur = head;
        int index = 1;
        ListNode pre = null;
        while (head != null) { 
            if (index <= m - 1) {
                if (index == m - 1) leftNode = cur;
                cur = cur.next;
            } else if (index >= m && index <= n) {
                if (index == m) reverseTail = cur;
                ListNode temp = cur.next;
                cur.next = pre;
                pre = cur;
                cur = temp;
                if (index == n) {
                    //此时cur已到第n+1个结点，需要存储起来
                    rightNode = cur;
                    //pre 在第n个结点
                    reverseHead = pre;
                    break;
                }
            }
            index++;
        }
        //开始连接
        if (leftNode != null) {
            leftNode.next = reverseHead;
        } else {
            head = reverseHead;
        }
        reverseTail.next = rightNode;
        return head;
    }