Leetcode 668, 719

668 找到乘法表里的第k小的树

 

显然是用二分，不过这个题目不是二分m和n的位置，而是二分候选答案，我们知道候选的是在1和m*n之间，对于每个值，我们可以在O(min(m, n))的时间内求出来有多少数比它小

时间复杂度是O(min(m, n)*log(m*n))

    public int findKthNumber(int m, int n, int k) {
        int l = 1;
        int r = m * n;
        if (m > n) {
            int t = m;
            m = n;
            n = t;
        }
        while (l < r) {
            int mid = (l + r) / 2;
            int rank = findRank(m, n, mid);
            if (rank < k) l = mid + 1;
            else r = mid;
        }
        return l;
    }

    //计算<=x的数有多少个
    int findRank(int m, int n, int x) {
        int cnt = 0;
        for (int i = 1; i <= m; i++) {
            cnt += Math.min(n, x / i);
        }
        return cnt;
    }

 

 

719：和上题类似

通过排序后，我们知道最少距离是0，最大距离是a[n-1]-a[0]

我们可以二分答案mid，然后判断<=mid的有多少个，

 

这里判断的时候考虑使用滑动窗口，枚举左端点start，然后end向右移动到第一个不满足 a[end] - a[start] <=k的位置，接下来start+1，end继续向右移动。

由于start和end都只循环了一次，因此时间复杂度是O(N)

加上二分，那么总的时间复杂度是O(N * log(a[n-1]-a[0]))

 

    public int smallestDistancePair(int[] nums, int k) {
        int len = nums.length;
        Arrays.sort(nums);
        
        int l = 0;
        int r = nums[len - 1] - nums[0];
        while (l < r) {
            int mid = (l + r) / 2;

            int cnt = 0;
            int end = 1;
            for (int start = 0; start < len; start++) {
                //第一个和num[start]的差>mid的时候退出循环
                while (end < len && nums[end] - nums[start] <= mid) end++;
                cnt += end - start - 1;
            }

            if (cnt < k) l = mid + 1;
            //因为会有并列的数
//            else if (cnt > k) r = mid;
//            else r = mid;
            //因此可以合并为一句话
            else r = mid;

        }
        return l;
    }

 

 

这两个题目有共同的地方都是枚举答案，也是二分的一个常见的应用。

还有就是因为有相同数字的存在，要注意二分的条件。