题目描述
给定一个包含 n + 1
个整数的数组 nums
,其数字都在 1
到 n
之间(包括 1
和 n
),可知至少存在一个重复的整数。假设只有一个重复的整数,找出这个重复的数。
示例 1:
输入: [1,3,4,2,2]
输出: 2
示例 2:
输入: [3,1,3,4,2]
输出: 3
说明:
不能更改原数组(假设数组是只读的)。
只能使用额外的
O
(
1
)
O(1)
O(1) 的空间。
时间复杂度小于
O
(
n
2
)
O(n^2)
O(n2) 。
数组中只有一个重复的数字,但它可能不止重复出现一次。
给定代码
class Solution {
public int findDuplicate(int[] nums) {
}
}
题解一
力扣大佬的二分法,实在是强,完全想不到。
我首先想到能像取次数为一的值
那种题一样,进行位运算行得通吗?然后发现行不通。
题目要求不能修改原数组,而且空间只能开常数。所以给数组排序、哈希记录次数等方法都不可取了。
再看题目要求,数组中的数为[1, n]
,那么最容易想到的一个
O
(
n
2
)
O(n^2)
O(n2)的解决办法是:针对每个数,遍历一次数组,如果数组中有相同的数,当前数就是重复的。
但题目要求时间复杂度小于 O ( n 2 ) O(n^2) O(n2),所以不能一个一个判断。
那只能一次判断一半了,有一个结论:
抽屉原理:桌上有十个苹果,要把这十个苹果放到九个抽屉里,无论怎样放,我们会发现至少会有一个抽屉里面放不少于两个苹果。
这里直接上例子:
若数组为[1, 4, 2, 5, 2, 3]
,数字在[1, 5]
之间。
取[1, 5]
的中间值3
,遍历数组统计小于等于3
的数有多少个,这里有[1, 2, 2, 3]
一共有四个。
说明[1, 3]
之间肯定有重复值。那么就能缩小搜索范围,继续搜下去,直到区间重合。
如果数组中只有一个值重复,数组中[1, m]之间的数的个数大于m,那么[1, m]之间一定有重复。否则,重复区间一定在[m + 1, n]中。
class Solution {
public int findDuplicate(int[] nums) {
int left = 1;
int right = nums.length - 1;
while (left < right)
{
int count = 0;
int medium = (left + right) / 2;
for (int num : nums)
{
if (num <= medium) {
count++;
}
}
if (count > medium) {
right = medium;
} else {
left = medium + 1;
}
}
return left;
}
}
时间复杂度: O ( n l o g ( n ) ) O(n log(n)) O(nlog(n))。二分法耗时 l o g ( n ) log(n) log(n),每次要完整遍历一次数组,耗时 O ( n ) O(n) O(n)。
空间复杂度: O ( 1 ) O(1) O(1)。