牛客网高频算法题系列-BM2-链表内指定区间反转

牛客网高频算法题系列-BM2-链表内指定区间反转

题目描述

将一个节点数为 size 链表 m 位置到 n 位置之间的区间反转，要求时间复杂度 O(n)，空间复杂度 O(1)。

原题目见：BM2 链表内指定区间反转

解法一：链表遍历，指针交换

因为起始位置可能是头结点，所以首先设置一个虚拟的头结点dummyNode并将next指向原有的头结点，然后处理过程如下：

• 首先遍历链表，找到起始位置m的前一个结点pre，用来记录反转前的结点；
• 然后用cur和next记录pre的next结点，用next记录cur的next结点；
• 然后继续遍历链表，通过交换pre、next、cur的next指针，将next结点转到pre结点的下一个结点处，然后循环处理cur的下一个结点；
• 遍历到结束结束位置n的结点即反转结束。
• 最后，返回dummyNode结点的next结点即为反转后的链表。

public class Bm002 {
    /**
     * @param head ListNode类
     * @param m    起始位置
     * @param n    结束位置
     * @return ListNode类
     */
    public static ListNode reverseBetween(ListNode head, int m, int n) {
        if (head == null || head.next == null) {
            return head;
        }
        if (m == n) {
            return head;
        }
        // 设置虚拟头结点
        ListNode dummyNode = new ListNode(-1);
        dummyNode.next = head;
        // pre为反转区间的前一个结点
        ListNode pre = dummyNode;
        for (int i = 0; i < m - 1; i++) {
            pre = pre.next;
        }
        // cur初始为反转区间的起始结点
        ListNode cur = pre.next;
        ListNode next;
        for (int i = 0; i < n - m; i++) {
            // 通过交换next指针指向，将next结点转到pre结点的下一个结点处
            next = cur.next;
            cur.next = next.next;
            next.next = pre.next;
            pre.next = next;
        }

        // 最后，返回dummyNode结点的next结点即为反转后的链表
        return dummyNode.next;
    }

    public static void main(String[] args) {
        ListNode head = ListNode.testCase2();
        System.out.println("指定区间反转之前");
        ListNode.print(head);

        ListNode newHead = reverseBetween(head, 2, 4);
        System.out.println("指定区间反转之后");
        ListNode.print(newHead);
    }
}

相信坚持的力量！