Question:
Suppose you have an array of 1001 integers. The integers are in random order, but you know each of the integers is between 1 and 1000 (inclusive). In addition, each number appears only once in the array, except for one number, which occurs twice. Assume that you can access each element of the array only once. Describe an algorithm to find the repeated number. If you used auxiliary storage in your algorithm, can you find an algorithm that does not require it?
如果可以用辅助空间的话,可以用HashSet来处理,如果add失败说明有重复的。但是如果不能用额外空间的话呢?
Solution 1:
考虑加法。1+2+...+n = n(n-1)/2
假设数组所有元素之和为sum, 那么sum-n(n-1)/2就是我们要找的重复的数。
public int findDuplicate(int[] nums) {
int sum = 0;
int n = nums.length;
int expected = n*(n-1)/2;
for(int i=0; i<n; i++) {
sum += nums[i];
}
return sum-expected;
}
但是数组的元素数达到或接近INT_MAX的时候, 加法运算就会溢出。我们还可以考虑位运算XOR。
XOR具有以下性质:
交换律:
结合律:
恒等律:
归零律:
自反:
假设给定数组A[6] = {5,2,3,4,3,1},而我们期望的数组应该为A[6]={0,1,2,3,4,5}。其中我们用0来代替重复的元素。
接下来对给定数组和我们期望数组的所有元素做异或运算:
(5^0)^(2^1)^(3^2)^(4^3)^(3^4)^(1^5)
= (5^5)^(2^2)^(3^3)^(4^4)^(1^1)^(3^0)
= 0^3
= 3
写成代码则非常简单。
Solution2:
public int findDuplicate(int[] nums) {
int dup = 0;
for(int i=0; i<nums.length; i++) {
dup ^= nums[i]^i;
}
return dup;
}
Reference: