给定一个包含 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, mid]
的数小于等于 mid,说明此范围内没有重复的数,否则说明有重复数。
class Solution:
def findDuplicate(self, nums: List[int]) -> int:
left, right = 1, len(nums) - 1
while left < right:
mid = (left + right) >> 1
cnt = sum(v <= mid for v in nums)
if cnt > mid:
right = mid
else:
left = mid + 1
return left
class Solution {
public int findDuplicate(int[] nums) {
int left = 1, right = nums.length - 1;
while (left < right) {
int mid = (left + right) >> 1;
int cnt = 0;
for (int v : nums) {
if (v <= mid) {
++cnt;
}
}
if (cnt > mid) {
right = mid;
} else {
left = mid + 1;
}
}
return left;
}
}
class Solution {
public:
int findDuplicate(vector<int>& nums) {
int left = 1, right = nums.size() - 1;
while (left < right) {
int mid = (left + right) >> 1;
int cnt = 0;
for (int& v : nums)
if (v <= mid)
++cnt;
if (cnt > mid)
right = mid;
else
left = mid + 1;
}
return left;
}
};
func findDuplicate(nums []int) int {
left, right := 1, len(nums)-1
for left < right {
mid := (left + right) >> 1
cnt := 0
for _, v := range nums {
if v <= mid {
cnt++
}
}
if cnt > mid {
right = mid
} else {
left = mid + 1
}
}
return left
}
/**
* @param {number[]} nums
* @return {number}
*/
var findDuplicate = function (nums) {
let left = 1,
right = nums.length - 1;
while (left < right) {
const mid = (left + right) >> 1;
let cnt = 0;
for (let v of nums) {
if (v <= mid) {
++cnt;
}
}
if (cnt > mid) {
right = mid;
} else {
left = mid + 1;
}
}
return left;
};
function findDuplicate(nums: number[]): number {
let left = 1,
right = nums.length - 1;
while (left < right) {
const mid = (left + right) >> 1;
let cnt = 0;
for (let v of nums) {
if (v <= mid) {
++cnt;
}
}
if (cnt > mid) {
right = mid;
} else {
left = mid + 1;
}
}
return left;
}