给定一个包含 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) 。
- 数组中只有一个重复的数字,但它可能不止重复出现一次。+
思路:
因为数据的个数是n+1个,取值是 1 到 n 之间。
我们使用二分查找,每次计算一个mid, 统计数组中,小于或等于mid的数字个数,如果个数大于mid个,那么就说明重复的那个数,一定在[l,mid]这个区间。
这样时间复杂度是O(nlogn)
代码:
class Solution {
public:
int findDuplicate(vector<int>& nums) {
int n = nums.size();
int l=1,r = n-1;
int ans = 0;
while(l<=r)
{
int mid= (l+r)/2;
//统计比mid小的数的数量
int count = 0;
for(int i=0; i<nums.size(); i++)
{
if(nums[i]<=mid)
count++;
}
if(count>mid)
{
ans=mid;//答案一定在[l,mid]之间,先用ans存下mid,再去区间[l,mid-1]之间寻找答案
r = mid-1;
}else
{
l=mid+1;
}
}
return ans;
}
};