Easy草稿★★★★★O(n) 時間 · O(1) 空間
ArrayHash Table
Patterns#️⃣ 雜湊
尚未複習過

解法已隱藏 — 先讀題目敘述、自己想想看,再點上方按鈕揭曉。

448Find All Numbers Disappeared in An ArrayArrays & HashingEasyArrays & Hashing

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

Example:

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

Intuition

TIP

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

Approaches

1. HashSet — O(n) / O(n)
  • Idea: 將所有元素放入 HashSet,遍歷 1..n 找不在 Set 中的
  • Time: O(n)
  • Space: 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. In-place Marking (negation) — O(n) / O(1)
  • Idea: 用元素值當索引,將對應位置標記為負數,最後找出仍為正數的位置
  • Time: O(n) - 兩次遍歷
  • Space: 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
    }
}
Note: 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