Leetcode287. 寻找重复数-----二分查找、数组重排序
题目表述
给定一个包含 n + 1 个整数的数组 nums ,其数字都在 [1, n] 范围内(包括 1 和 n),可知至少存在一个重复的整数。
假设 nums 只有 一个重复的整数 ,返回 这个重复的数 。
你设计的解决方案必须 不修改 数组 nums 且只用常量级 O(1) 的额外空间。
示例:
输入:nums = [1,3,4,2,2]
输出:2
二分查找
抽屉原理,把10个苹果放进9个抽屉,至少有一个抽屉里至少放2个苹果。
「二分查找」的思路是先猜一个数(搜索范围 [left..right]里位于中间的数 mid),然后统计原始数组中小于等于mid的元素的个数 count:
- 如果 count 严格大于 mid。根据 抽屉原理,重复元素就在区间 [left..mid] 里;
- 否则,重复元素可以在区间 [mid + 1..right] 里找到,其实只要第 1 点是对的,这里肯定是对的,但要说明这一点,需要一些推导,我们放在最后说。
class Solution {
public int findDuplicate(int[] nums) {
int l = 0;
int r = nums.length - 1;
while(l <= r){
int mid = l + (r - l) / 2;
int count = 0;
for(int num : nums){
if(num <= mid){
count++;
}
}
if(count > mid){
r = mid - 1;
}else{
l = mid + 1;
}
}
return l;
}
}
数组重排序(原地交换)
和Leetcode442、 剑指 Offer 03. 数组中重复的数字、41. 缺失的第一个正数题目类似
class Solution {
public int findDuplicate(int[] nums) {
int n = nums.length;
int i = 0;
while(i < n){
if(nums[i] == i){
i++;
continue;
}
if(nums[nums[i]] == nums[i]) return nums[i];
int tmp = nums[i];
nums[i] = nums[nums[i]];
nums[tmp] = tmp;
}
return -1;
}
}