原题地址:https://leetcode-cn.com/problems/find-the-duplicate-number/submissions/
题目描述:
给定一个包含 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) 。
数组中只有一个重复的数字,但它可能不止重复出现一次。
解题方案:
题型是查找题型,不能对原数据进行修改。本题利用的是二分搜索法:
首先计算mid,然后统计数组中小于等于mid元素的个数k。如果k <= mid的话,那么说明重复值在[mid+1,n]之间,否则的话重复值在[1,mid]之间。为什么这样子做是合理的呢?实际上这个问题的思想来源于抽屉原理,我们将包含重复元素的所有数放到一个可以容纳所有数的数组中去。
妙啊,这是抽屉原理问题啊!
代码:
class Solution {
public:
int findDuplicate(vector<int>& nums) {
int low = 1, high = nums.size() - 1;
while (low < high)
{
int mid = (high + low)/2;
int count = 0;
for (auto num : nums)
if (num <= mid)
++count;
if (count <= mid)
low = mid + 1;
else
high = mid;
}
return low;
}
};
另一种解题方法,也是很好的,不过已经对原数据进行修改了:
class Solution {
public:
int findDuplicate(vector<int>& nums) {
for(int i=0;i<nums.size();++i){
int idx=abs(nums[i])-1;
if(nums[idx]<0) return idx+1;
nums[idx]=-nums[idx];
}
}
};
都是值得学习的~~