BJFUOJ-418

查找数组中缺失的数字

#include<iostream>
#include<vector>
using namespace std;

//参考leetcode 448题解
//使用哈希表，用来记录[1,n]的元素是否出现,但是空间复杂度是O(N)
//空间复杂度优化为O(1)，可以使用原地哈希表，使用原数组充当哈希表
//数组中的数的范围为[1,n]
//利用[1,n]范围之外的数字，来表达是否存在
//映射到哈希表 0 - n-1
vector<int> findDisappearedNumbers(vector<int> &v) {
	int n = v.size();
	for (auto num : v) {

		int x = (num - 1) % n; //若当前值被标记过，需要取模来取出它原本的值
		
		v[x] += n;  //每次遍历到一个值，就将该值对应的哈希位置进行标记

	}
	//因为规定了每个元素的值在[1,n]
	//最后没有被标记过的，也就是没有+=n的位置，就是没有出现过的数字
	vector<int>res;
	for (int i = 0; i < n; i++) {
		if (v[i] <= n) {
			res.push_back(i + 1);
		}
	}
	return res;
}

int main() {
	int n;
	while (cin >> n && n!=0) {
		vector<int>v;
		for (int i = 0; i < n; i++) {
			int x; cin >> x;
			v.push_back(x);
		}
		vector<int> res = findDisappearedNumbers(v);
		if (!res.empty()) {
			for (int i = 0; i < res.size(); i++) {
				cout << res[i];
				if (i != res.size() - 1)cout << " ";
			}
		}
		else {
			cout << "Not Found";
		}
		
		cout << endl;
	}

	
	return 0;
}