leetcode原题链接:寻找重复数
上一篇:HOT99-下一个排列
下一篇:当前已经是最后一篇啦
题目描述
给定一个包含 n + 1
个整数的数组 nums
,其数字都在 [1, n]
范围内(包括 1
和 n
),可知至少存在一个重复的整数。假设 nums
只有 一个重复的整数 ,返回 这个重复的数 。你设计的解决方案必须 不修改 数组 nums
且只用常量级 O(1)
的额外空间。
示例 1:
输入:nums = [1,3,4,2,2] 输出:2
示例 2:
输入:nums = [3,1,3,4,2] 输出:3
提示:
1 <= n <= 105
nums.length == n + 1
1 <= nums[i] <= n
nums
中 只有一个整数 出现 两次或多次 ,其余整数均只出现 一次
进阶:
- 如何证明
nums
中至少存在一个重复的数字? - 你可以设计一个线性级时间复杂度
O(n)
的解决方案吗?
解题方法:二分查找。本题我们换一种视角,我们在[1,2,3,4,...n]中二分查找目标数target,这个目标数target在数组nums中必然满足: nums数组中小于或等于target的数字的个数必然大于target。比如:
[1, 2, 2, 3, 4] ==> cnt[<=1] = 1, cnt[<=2]=3, cnt[<=3]=4, cnt[<=4]=5
显然对于target=2, 3, 4,都满足要求,而我们要查找的结果是2。
换句话说:本题可以改为,从[1, n]中查找满足条件的最小k,满足的条件:k在nums中出现的次数>=k。
C++代码
// 写法1: 循环执行条件 left < right,最终返回left(或者right)指向的位置
#include <iostream>
#include <vector>
class Solution {
public:
int findDuplicate(const std::vector<int>& nums) {
int n = nums.size();
int left = 1;
int right = n;
while (left < right) {
int mid = (left + right) / 2;
//判断mid值是否满足在nums中cnt <= num
int cnt = 0; //记录nums中 <= mid的值的个数
for (auto num : nums) {
if (num <= mid) {
cnt++;
}
}
if (cnt > mid) { //满足重复个数的要求
right = mid; //寻找[left, right]中满足要求的最小数, 最后right指向目标值
} else {
left = mid + 1;
}
}
return right;//最终left和right指向同一个元素,此处也可以返回left
}
};
// 写法2:循环执行条件left <= right, 需要用一个单独变量保存解
#include <iostream>
#include <vector>
class Solution {
public:
int findDuplicate(const std::vector<int>& nums) {
int n = nums.size();
int left = 1;
int right = n;
int result = -1;
while (left <= right) {
int mid = (left + right) / 2;
//判断mid值是否满足在nums中cnt <= num
int cnt = 0; //记录nums中 <= mid的值的个数
for (auto num : nums) {
if (num <= mid) {
cnt++;
}
}
if (cnt > mid) { //满足重复个数的要求
right = mid - 1; //寻找[left, right]中满足要求的最小数, 最后right指向目标值
result = mid;
} else {
left = mid + 1;
}
}
return result;
}
};