Description

給定長度為 n 的陣列 nums,其中每個元素在 [1, n] 範圍內。回傳所有在 [1, n] 中但未出現在 nums 中的數字。要求 O(n) 時間且不使用額外空間(輸出不計)。

Example:

Input: nums = [4,3,2,7,8,2,3,1] Output: [5,6]

Intuition#

核心思路:利用陣列本身作為 Hash Table——將出現過的數字對應位置標記為負數,最後正數位置就是缺失的數字。

  • 數字範圍是 [1, n],恰好對應陣列索引 [0, n-1]
  • 遍歷陣列,對每個值 |nums[i]|,將 nums[|nums[i]| - 1] 設為負數
  • 第二次遍歷,正數位置的索引 + 1 就是缺失的數字

Approaches#

1: HashSet#

  • 概念: 將所有元素放入 HashSet,遍歷 1..n 找不在 Set 中的
  • 時間複雜度: O(n)
  • 空間複雜度: O(n) - HashSet
class Solution {
    fun findDisappearedNumbers(nums: IntArray): List<Int> {
        val set = nums.toHashSet()
        val result = mutableListOf<Int>()
        for (i in 1..nums.size) {
            if (i !in set) result.add(i)
        }
        return result
    }
}

⭐2: 原地標記(Negation)#

  • 概念: 用元素值當索引,將對應位置標記為負數,最後找出仍為正數的位置
  • 時間複雜度: O(n) - 兩次遍歷
  • 空間複雜度: O(1) - 原地修改(不計輸出)
class Solution {
    fun findDisappearedNumbers(nums: IntArray): List<Int> {
        // 標記出現過的數字
        for (num in nums) {
            val idx = Math.abs(num) - 1
            if (nums[idx] > 0) {
                nums[idx] = -nums[idx]
            }
        }
        // 收集未被標記的位置
        val result = mutableListOf<Int>()
        for (i in nums.indices) {
            if (nums[i] > 0) {
                result.add(i + 1)
            }
        }
        return result
    }
}
補充:Cyclic Sort

將每個數字放到正確位置(nums[i] 應該在索引 nums[i]-1),最後不在正確位置的就是缺失的。

class Solution {
    fun findDisappearedNumbers(nums: IntArray): List<Int> {
        var i = 0
        while (i < nums.size) {
            val correct = nums[i] - 1
            if (nums[i] != nums[correct]) {
                nums[i] = nums[correct].also { nums[correct] = nums[i] }
            } else {
                i++
            }
        }
        val result = mutableListOf<Int>()
        for (j in nums.indices) {
            if (nums[j] != j + 1) result.add(j + 1)
        }
        return result
    }
}

🔑 Takeaways#

  • Pattern: 當數字範圍在 [1, n] 且陣列長度為 n 時,可以用「原地標記」或「Cyclic Sort」達到 O(1) 額外空間
  • 關鍵技巧: 取絕對值避免重複標記導致錯誤;此模式適用於 LC 442(找重複)、LC 41(找缺失正整數)等題