题目
给定一个范围在 1 ≤ a[i] ≤ n ( n = 数组大小 ) 的 整型数组,数组中的元素一些出现了两次,另外一些只出现一次。数组
找到全部在 [1, n] 范围之间没有出如今数组中的数字。网络
您能在不使用额外空间且时间复杂度为O(n)的状况下完成这个任务吗? 你能够假定返回的数组不算在额外空间内。app
示例:ide
输入:
[4,3,2,7,8,2,3,1]spa
输出:
[5,6]code
来源:力扣(LeetCode)
连接:https://leetcode-cn.com/problems/find-all-numbers-disappeared-in-an-array
著做权归领扣网络全部。商业转载请联系官方受权,非商业转载请注明出处。blog
看到这个题的第一反应,利用额外vector空间,使用nums中的元素做其下标,对索引到的位置染色/作标记。最后从位置1开始(1 ≤ a[i] ≤ n),遍历一下额外的vector,没有被染色/标记的位置下标,就是nums中不存在的元素。
但题目要求不能有额外的空间,而且时间复杂度O(n),那只能让nums本身卷起来了,只能在nums自己上利用上述关系了。索引
可经过如下方式进行染色/标记:leetcode
class Solution {
public:
vector<int> findDisappearedNumbers(vector<int>& nums) {
std::vector<int> vecRet;
for(const auto& elem : nums){
if(elem < 0){ //elem所在位置已染色
if(nums[-1 * elem - 1] < 0){ //(-1 * elem - 1)位置已被染色,无需再次染色(nums中elem重复出现)
continue;
}
nums[-1 * elem - 1] *= -1; //使用elem的原值-1做下标继续染色,来代表elem的原值存在
}
else{
if(nums[elem - 1] < 0){ //(elem - 1)位置已被染色,无需再次染色(nums中elem重复出现)
continue;
}
nums[elem - 1] *= -1; //1 ≤ a[i] ≤ n,用(elem-1)做下标,对下标位置染色,来代表elem存在
}
}
for(int i = 0; i < nums.size(); i++){
if(nums[i] > 0){
vecRet.push_back(i + 1); //1 ≤ a[i] ≤ n,下标为elem - 1,求原值需+1
}
}
return vecRet;
}
};