题目
给定一个包含 n + 1 个整数的数组 nums,其数字都在 1 到 n 之间(包括 1 和 n),可知至少存在一个重复的整数。假设只有一个重复的整数,找出这个重复的数。
示例1
输入: [1,3,4,2,2]
输出: 2
示例2
输入: [3,1,3,4,2]
输出: 3
说明
- 不能更改原数组(假设数组是只读的)。
- 只能使用额外的 O(1) 的空间。
- 时间复杂度小于 O(n2) 。
- 数组中只有一个重复的数字,但它可能不止重复出现一次。
思路
- 假如数组中的数在1~n。只有一对重复的数,其他都是不重复的。left = 1,right = n。
- 计算mid = (1+n) / 2。遍历一遍数组,计算数组中小于等于mid的数有temp个。如果temp大于mid,说明说明重复的数字在包括mid的左边,此时让right = mid;否则重复的数字在mid右边,让left = mid + 1。
- 重复上述过程,直到left = right。此时找到了重复的数字。left, right表示数组中最后两个数,mid = left, 如果小于等于left的数大于1,则right = mid,跳出循环。如果大于left的数大于1个,则left = mid + 1,跳出循环。
代码
class Solution {
public:
int findDuplicate(vector<int>& nums) {
int left = 1;
int right = nums.size() - 1;
while ( left < right ) {
int mid = ( left + right ) >> 1;
int temp = 0;
for ( int i = 0; i < nums.size(); ++i ) {
if ( nums[i] <= mid )
++temp;
}
if ( temp <= mid )
left = mid + 1;
else
right = mid;
}
return left;
}
};