题目地址:
https://www.lintcode.com/problem/find-all-numbers-disappeared-in-an-array/description
给定一个长 n n n且只含 1 ∼ n 1\sim n 1∼n数字的数组 A A A,其中有两个数字出现了两次,其他数字只出现一次。要求求出 1 ∼ n 1\sim n 1∼n未出现的数字。
思路是,将所有出现的数尽可能交换到其应该在的位置上,例如 3 3 3应该在 A [ 2 ] A[2] A[2]这个地方。然后再扫描一遍,找到未出现的数即可。具体请看代码,代码如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
/**
* @param nums: a list of integers
* @return: return a list of integers
*/
public List<Integer> findDisappearedNumbers(int[] nums) {
// write your code here
List<Integer> res = new ArrayList<>();
// 先尽可能将数字归位
for (int i = 0; i < nums.length; i++) {
while (nums[nums[i] - 1] != nums[i]) {
swap(nums, nums[i] - 1, i);
}
}
// 然后找出位置上放的数不对的那些位置,应该放的数就是未出现的数
for (int i = 0; i < nums.length; i++) {
if (nums[i] != i + 1) {
res.add(i + 1);
}
}
return res;
}
private void swap(int[] nums, int i, int j) {
int tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。