No.287 Leetcode题目 “寻找重复数”

题目描述

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

示例 1:

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

示例 2:

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

说明：

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

解决方法

1、二分查找法

由于题目要求时间复杂度小于O(n^2)，所以一般的暴力破解是不可以的，我们使用二分查找。
我们用题目中的数组[1 3 4 2 2]为例，设cnt值为小于等于数组中各元素的值。可见cnt值是递增的，我们只需要用二分法找到画红框的nums即可。
代码具体思路参考注释即可。

完整代码如下：

class Solution {
public:
    int findDuplicate(vector<int>& nums) {
        //利用二分法查找重复的整数
        int len = nums.size();
        int left = 1; //二分法左起点
        int right = len - 1; //二分法右起点，最后的答案肯定在left和right之间
        int answer = -1; //初始化最后答案

        // 开始查找
        while(left <= right){
          int mid = (left + right) / 2;
          int count = 0; //nums中小于mid的个数

          // 计算nums中小于mid的整数个数
          for(int i = 0;i < len; i++){
            if(nums[i] <= mid)
            	count++;
          } 

          // 如果小于等于mid的个数比mid小或与mid相等，则在右半部分继续查找
          if(count <= mid){
            left = mid + 1;
          }

          // 如果小于等于mid的个数比mid大，则在左半部分继续查找
          if(count > mid){
            right = mid - 1;
            answer = mid;
          }
        }
        return answer;
    }  
};