给定一个包含 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) 。
- 数组中只有一个重复的数字,但它可能不止重复出现一次。
//抽屉原理,查找哪边的苹果的数量大于了抽屉的数量
class Solution {
public:
int findDuplicate(vector<int> &nums) {
int n = nums.size() - 1;
int l = 1, r = n;//对1~n的数进行二分,而不是对nums中数二分
while (l < r) {
int mid = (l + r) >> 1;
int cnt = 0;
for (auto num : nums) cnt += (num >= l && num <= mid);
//如果小于等于左边的抽屉数量(mid - l + 1),说明左边一定没有重复的,答案在右边
if (cnt <= mid - l + 1) l = mid + 1;
else r = mid;
}
return l;
}
};