「每日一題」與面試官手撕代碼：如何科學高效的尋找重復元素？

public int getResult(int[] nums) {
    Set<Integer> set = new HashSet<Integer>();
    int result = -1;
    for (int num : nums) {
        if (!set.add(num)) {
            result = num;
            break;
        }
    }
    return result;
}

如果上面代碼格式出現問題，可以查看下面代碼圖片

找出任意一個重復數字

遍歷 n 次並且 Set 內容最多是 n 個字符，復雜度都是 O(n)

2. 找出任意一個重復數字

給定一個長度為 n 的數組 nums，判斷是否有重復值,並且兩個重復值距離不超過 k。

示例：輸入[1,2,3,2,1,4,5] ，k = 2 輸出 true(兩個 2)

思路：同問題一，只要多判斷一次 Set 長度即可。

題解：

public boolean containsNearbyDuplicate(int[] nums, int k) {
    Set<Integer> set = new HashSet<Integer>();
    boolean result = false;
    for (int i = 0; i < nums.length; i++) {
        if (!set.add(nums[i])) {
            result = true;
            break;
        }

        if (set.size() > k) { // 超過長度就刪除最遠的一個數
            set.remove(nums[i-k]);
        }
    }
    return result;
}

如果上面代碼格式出現問題，可以查看下面代碼圖片

找出任意一個重復數字

遍歷 n 次並且 k 最多 n 個字符，復雜度都是 O(n)

3. 尋找重復數

給定一個包含 n + 1 個整數的數組 nums，其數字都在 1 到 n 之間（包括 1 和
n），只有一個重復的整數，找出這個重復的數字。

示例：輸入[1,3,4,2,2] 輸出 2

思路：可通過問題 1 方式解決。還可以通過快慢指針法解決：將數組看成一個鏈表，下標為當前指針（node），值指向下一指針(nextNode)，數組出現重復的數字意味着有兩個指針的 nextNode 相同。然后通過快慢指針法解決。

題解：

public int getResult(int[] nums) {
    int result = 0;
    int slow = 0, fast = 0;
    do  {
        slow = nums[slow];
        fast = nums[nums[fast]];
    } while (slow != fast);

    do {
        slow = nums[slow];
        result = nums[result];
    } while (result != slow);
    return result;
}

如果上面代碼格式出現問題，可以查看下面代碼圖片

找出任意一個重復數字

遍歷 2n 次時間復雜的是 O(n)，只用了常量個字符，空間復雜度是 O(1)

4. 只出現一次的數

給定一個數組
nums，除了某個元素只出現一次以外，其余每個元素均出現兩次。找出那個只出現了一次的元素。

示例：輸入[1,1,4,2,2] 輸出 4

思路：可通過問題 1 方式解決。還可以通過位運算解決，兩個相同數異或后為 0，數組所有元素執行一次異或操作，剩下就是出現一次的數。

題解：

public int getResult(int[] nums) {
    int result = 0;
    for (int num : nums) {
        result = result ^ num;
    }
    return result;
}

如果上面代碼格式出現問題，可以查看下面代碼圖片

找出任意一個重復數字

遍歷 n 次時間復雜的是 O(n)，只用了常量個字符，空間復雜度是 O(1)

5. 只出現一次的數

給定一個數組
nums，除了兩個元素只出現一次以外，其余每個元素均出現兩次。找出那個只出現了一次的元素。

示例：輸入[1,1,3,4,2,2] 輸出[3,4]

思路：可通過問題 1 方式解決。還可以通過位運算解決，和題 4 區別是存在 2 個只出現一次的數，所以要想辦法把這兩個數區分出來。先將數組所有元素異或，得出的值是兩個只出現一次元素 a,b 的異或值 numsXOR（如 10010）。numsXOR 二進制中 1 的位就是 a 和 b 差異位（因為不同的值異或才是 1），現在只需要找 lowbit(最右一位差異值，10)，然后通過 lowbit 和 a
, b
進行與運算（&），得出的值就一定不同，這樣可以分出 a 和 b，最后按照異或運算就能得出結果（其他重復的不用管，不管分到哪一組，重復的數異或都是 0）

題解：

public int[] singleNumber(int[] nums) {
    int[] results = new int[]{0,0};
    int numsXOR = 0; // 兩個數異或值
    for (int num : nums) {
        numsXOR = numsXOR ^ num;
    }

    int lowBit = numsXOR & (-numsXOR); // lowbit值，用於區分a和b

    for (int num : nums) {
        if ((lowBit & num) == 0) {
            results[0] = results[0] ^ num;
        } else {
            results[1] = results[1] ^ num;
        }
    }

    return results;
}

如果上面代碼格式出現問題，可以查看下面代碼圖片

找出任意一個重復數字