剑指 Offer 03. 数组中重复的数字题解

题目链接

题目大意：有一个长度为 $n$ 的数组 nums 包括数字 $0\sim n-1$，现让找出任意重复的数字。

我最开始的思路是创建一个 set，将每个数字插入到 set 中，如果 set 长度不发生变化，则说明当前数字是重复数字。

代码：

#include <bits/stdc++.h>
using namespace std;

class Solution {
public:
    int findRepeatNumber(vector<int>& nums) {
        set<int> s;
        int len = 0;
        for (auto i : nums) {
            s.insert(i);
            if (s.size() == len) {
                return i;
            }
            len++;
        }
    }
};

int main(void) {
    Solution s;
    vector<int> nums{2, 3, 1, 0, 2, 5, 3};
    cout << s.findRepeatNumber(nums) << endl;
    return 0;
}

后来看到一份题解，发现可以进行原地置换，将空间复杂度从 $O(n)$ 优化到 $O(1)$。

因为题目中给出数组 nums 只包含 $0\sim n-1$ 的数字，所以可以将每一个数字放置在对应的索引下，即让 nums[n] = n，如果一个位置出现两个数字，则表示它是重复的，这与哈希的思路有些相似。

具体思路：从索引 0 处开始，将它的值 nums[0] 与对应位置的值 nums[nums[0]] 交换，直到当前位置恰好是对应的值，即 nums[0] = 0，然后再处理下一位直到数组末尾。

代码：

#include <bits/stdc++.h>
using namespace std;

class Solution {
public:
    int findRepeatNumber(vector<int>& nums) {
        int pos = 0;
        while (pos < nums.size()) {
            while (nums[pos] != pos) {
                if (nums[pos] == nums[nums[pos]]) {
                    return nums[pos];
                }
                swap(nums[pos], nums[nums[pos]]);
            }
            pos++;
        }
        return -1;
    }
};

int main(void) {
    Solution s;
    vector<int> nums{2, 3, 1, 0, 2, 5, 3};
    cout << s.findRepeatNumber(nums) << endl;
    return 0;
}