二分搜索解LeetCode 287寻找重复数

287. 寻找重复数

给定一个包含 n + 1 个整数的数组 nums,其数字都在 1 到 之间(包括 1 和 n),可知至少存在一个重复的整数。假设只有一个重复的整数,找出这个重复的数。

示例 1:

输入: [1,3,4,2,2]

输出: 2

示例 2:

输入:[3,1,3,4,2]

输出:3

说明:

  1. 不能更改原数组(假设数组是只读的)。
  2. 只能使用额外的 O(1) 的空间。
  3. 时间复杂度小于 O(n2) 。
  4. 数组中只有一个重复的数字,但它可能不止重复出现一次。+

思路:

因为数据的个数是n+1个,取值是 1 到 之间。

我们使用二分查找,每次计算一个mid, 统计数组中,小于或等于mid的数字个数,如果个数大于mid个,那么就说明重复的那个数,一定在[l,mid]这个区间。

这样时间复杂度是O(nlogn)

代码:

class Solution {
public:
    int findDuplicate(vector<int>& nums) {
		int n = nums.size();
		int l=1,r = n-1;
		int ans = 0;
		while(l<=r)
		{
			int mid= (l+r)/2;
			//统计比mid小的数的数量 
			int count = 0;
    		for(int i=0; i<nums.size(); i++)
    		{
    			if(nums[i]<=mid)
    				count++;
    		}
    		if(count>mid)
			{
				ans=mid;//答案一定在[l,mid]之间,先用ans存下mid,再去区间[l,mid-1]之间寻找答案 
				r = mid-1;
			}else
			{
				l=mid+1;	
			}
		}
		return ans;
    }
   
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章