leetcode学习日记9-寻找重复数

题目描述

给定一个包含 n + 1 个整数的数组 nums，其数字都在 1 到 n 之间（包括 1 和 n），可知至少存在一个重复的整数。假设只有一个重复的整数，找出这个重复的数。
输入: [1,3,4,2,2]
输出: 2
说明：
不能更改原数组（假设数组是只读的）。
只能使用额外的 O(1) 的空间。
时间复杂度小于 O(n2) 。
数组中只有一个重复的数字，但它可能不止重复出现一次。

初步探索

首先，最简单的一种思路是两两比较是否相等，如果相等就输出。然而，由于O(n²) 的限制，不能用这种算法。

其次，又想到可以利用这些数字都是1到n中，可以将他们相加，然后通过和1到n的总和比较得出相差多少。然而，这个算法也又缺陷，因为由于数组中重复的数字可能重复多次，难以做到准确判断。

最后，由于时间复杂度的限制，想到可能可以使用二分搜索法。但是不能进行排序，因为原数组不可改变，也不能使用O(n)的额外空间。

综合方法

如何将这些思考综合起来呢？
我想到即使没有排序，由于数字的范围是固定的，可以使用一种变相的二分法。模糊的思路是，如果这个数组是恰好1，2，3…n，那么这个数组可以看成是"平衡"的，因为每个数字都出现了一次，其中位数是在(1+n)/2的地方。然而，如果某一个数字出现了两次，那么会有两种情况

• 这个数字小于中位数，那么我们将数组排列后，新的中位数将会小于原本的中位数
• 这个数字大于中位数，则同理，新的中位数将大于原本的中位数
• 这个数字等于中位数，则中位数重复

是不是有点像二分法？这里，我们并不需要排序数组，只需要判断理论和实际的中位数的位置就可以逐步缩小搜索范围了。

• 如果新的中位数小于原本中位数，则重复数字小于原本中位数
• 同理可得新的中位数大于原本中位数的情况
• 中位数重复，那么重复的数字就是是中位数

如何找中位数的位置呢？我的方法是

1. 找到理论中位数，即左边界l和右边界r的平均, mid=(l+r)/2
2. 遍历数组，统计小于中位数的数字个数 smaller，大于中位数的数字个数 bigger，等于中位数的数字个数equal
3. 如果equal>2，说明mid就是我们要找的数字
4. 如果smaller > bigger, 则将搜索范围缩小到 l到mid，即将r赋值为floor(mid)，smaller<bigger同理
5. 重复1-4

细节

注意，在数学定义中，如果有偶数个数字，中位数则是中间两个数字的平均数。在这道题中，也必须延用这样的定义，而不能简单把mid定义为int，使其向下取整。与一般二分搜索不同的是，我们需要考虑大于和小于的数字个数。如果数列有偶数个数字，向下取整后的中位数会使得一个已经平衡的数列变得不平衡，因为原本小于中位数的数字被算作等于了。所以，最简单的解决方法就是使用double定义mid。当然，这个中位数的0.5在边界时是没有意义的，故在缩小搜索范围时可以用floor和ceiling函数变为整数。

代码

class Solution {
public:
    int findDuplicate(vector<int>& nums) {
        int n=nums.size()-1;
        int l=1;// left boundary
        int r=n;// right boundary: both are representing numbers but not indexes
        int ans=0;//solution
        while(1){
            int bigger=0;//the count of numbers bigger than the median
            int smaller=0;// the count of numbers smaller than the median
            int equal=0;// the count of numbers equal to median
            
            double mid=(l+r)*0.5;//calculate median
            int i=0;
            for(i=0;i<=n;i++)
            {
                if(nums[i]>mid&&nums[i]<=r) bigger++;
                if(nums[i]<mid&&nums[i]>=l) smaller++;
                if(nums[i]==mid) equal++;
            }
            if(equal>=2){
                ans=mid;// answer found!
                break;
            }
            if(bigger>smaller) l=ceil(mid);// modify the boundary
            if(bigger<=smaller) r=floor(mid);
            
        }
        return ans;
    }
};