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

雄獅虎豹 發表於 2022-06-02
演算法

牛客網高頻演算法題系列-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$
相信堅持的力量!

相關文章