牛客網高頻演算法題系列-BM8-連結串列中倒數最後k個結點

牛客網高頻演算法題系列-BM8-連結串列中倒數最後k個結點

題目描述

描述：輸入一個長度為 n 的連結串列，設連結串列中的元素的值為 a~i~ ，返回該連結串列中倒數第k個節點。如果該連結串列長度小於k，請返回一個長度為 0 的連結串列。

原題目見：BM8 連結串列中倒數最後k個結點

解法一：雙指標法

首先，考慮兩種特殊情況：

• 如果原連結串列為空，直接返回null。
• 如果k不是正數，直接返回null。

否則，使用雙指標求解，求解過程如下：

• 首先，遍歷連結串列，將fast結點指向第k個結點；
• 如果遍歷完後fast為null，說明連結串列長度小於k，不存在倒數第k個結點，直接返回null；
• 否則，快慢指標一起移動，直到fast移動到最後一個結點，此時，slow即為倒數第k個結點，返回之。

程式碼

public class Bm008 {
    /**
     * 連結串列中倒數最後k個結點
     *
     * @param pHead ListNode類
     * @param k     int整型
     * @return ListNode類
     */
    public static ListNode findKthToTail(ListNode pHead, int k) {
        // 如果原連結串列為空，直接返回null
        if (pHead == null) {
            return null;
        }
        // 如果k不是正數，直接返回null
        if (k <= 0) {
            return null;
        }
        ListNode fast = pHead;
        // 將fast結點指向第k個結點
        for (int i = 0; i < k - 1 && fast != null; i++) {
            fast = fast.next;
        }
        // 如果fast為null，說明連結串列長度小於k，不存在倒數第k個結點，直接返回null
        if (fast == null) {
            return null;
        }

        ListNode slow = pHead;
        // 快慢指標一起移動，直到fast移動到最後一個結點，此時，slow即為倒數第k個結點
        while (fast.next != null) {
            slow = slow.next;
            fast = fast.next;
        }

        return slow;
    }

    public static void main(String[] args) {
        ListNode pHead = ListNode.testCase2();
        System.out.println("原連結串列為");
        ListNode.print(pHead);
        int k = 2;
        System.out.println("連結串列中的倒數第" + k + "個結點為：");
        System.out.println(findKthToTail(pHead, k));
    }
}

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