Leetcode 448 - Find All Numbers Disappeared in an Array （Array）

题意

给一个大小为n的数组a，数组内的元素范围为[1, n]。

已知数组内有些元素出现了两次或1次，求数组内没有出现的元素。

要求：

1. $O(n)$时间
2. 无额外空间。
3. 保存结果的那个vector不算额外空间。

思路

非常trick的一道题啊。

$O(n)$的时间就排除了排序，虽然计数排序可以，但是好像会用额外空间？

然后无额外空间的话，我们只能利用现有空间了。

那么，假如我们可以用额外空间的话，那么一个比较直观的思路就是我们开一个数组vis[]，假设一个元素x在a[]中出现过，那么我们就另vis[x] = 1。最后，统计一下vis[y] = 0的那些值就可以了。

既然我们不能用额外空间，那么，我们就要想办法利用原来的数组来达到vis的功能。我们想，假如在x在a[]中出现了，现在我们如果另a[x] = 1的话，就会破坏掉a[x]处本身的值。好了，既然我们不能破坏a[x]本身的值，还要记录a[x]出现过，并且题意告诉我们所有数都为正数并且[1,n]这个范围内。那如果我们另a[x]处的值变为负数，就可以避免a[x]处的值被破坏，还能记录a[x]的值出现过。

于是就可以得到我们的算法：

1. 遍历数组a里面的元素，假设当前元素为x，那么我们令a[abs(x) - 1]处的值变为本身的负数。
2. 再次遍历数组，其中为正数的值的下标+1代表没有出现过的数。

代码

class Solution {
public:
    vector<int> findDisappearedNumbers(vector<int>& nums) {
        vector<int> ans;
        for (auto x : nums) nums[abs(x) - 1] = -abs(nums[abs(x) - 1]);
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] > 0) ans.push_back(i + 1);
        }
        return ans;
    }
};