牛客網高頻演算法題系列-BM3-連結串列中的節點每k個一組翻轉

牛客網高頻演算法題系列-BM3-連結串列中的節點每k個一組翻轉

題目描述

將給出的連結串列中的節點每 k 個一組翻轉，返回翻轉後的連結串列如果連結串列中的節點數不是 k 的倍數，將最後剩下的節點保持原樣你不能更改節點中的值，只能更改節點本身。

原題目見：BM3 連結串列中的節點每k個一組翻轉

解法一：遞迴

使用遞迴的方式處理每k個一組的結點，具體處理方式如下：

• 首先用tail記錄每次翻轉的尾部，就是第k+1個結點，從頭結點開始，遍歷連結串列，找到第k+1個結點，如果當前結點數不夠k個，則不用反轉直接返回；
• 然後從連結串列頭開始遍歷，翻轉結點並記錄反轉後的表頭pre和表尾結點head；
• 然後遞迴處理從tail開始的後面的連結串列結點，並將head指向遞迴處理後的連結串列；
• 返回新的表頭即為反轉後的新的連結串列的表頭結點。

程式碼

public class Bm003 {
    /**
     * 遞迴
     *
     * @param head ListNode類
     * @param k    int整型
     * @return ListNode類
     */
    public static ListNode reverseKGroup(ListNode head, int k) {
        // 找到每次翻轉的尾部
        ListNode tail = head;
        // 遍歷k次到尾部
        for (int i = 0; i < k; i++) {
            // 如果不足k到了連結串列尾部，直接返回，不翻轉
            if (tail == null) {
                return head;
            }
            tail = tail.next;
        }
        // 翻轉時需要的前序和當前節點
        ListNode pre = null, cur = head;
        // 在到達當前段尾部節點前
        while (cur != tail) {
            ListNode temp = cur.next;
            cur.next = pre;
            pre = cur;
            cur = temp;
        }
        // 當前尾部指向下一段要翻轉的連結串列
        head.next = reverseKGroup(tail, k);
        return pre;
    }

    public static void main(String[] args) {
        ListNode head = ListNode.testCase2();
        System.out.println("翻轉之前");
        ListNode.print(head);

        // 當k=2時
        ListNode newHead = reverseKGroup(head, 2);
        System.out.println("當k=2時，翻轉之後");
        ListNode.print(newHead);

        ListNode head1 = ListNode.testCase2();
        // 當k=3時
        ListNode newHead2 = reverseKGroup(head1, 3);
        System.out.println("當k=3時，翻轉之後");
        ListNode.print(newHead2);
    }
}

$1.01^{365} ≈ 37.7834343329$
$0.99^{365} ≈ 0.02551796445$
相信堅持的力量！