LeetCode668.乘法表中第 k 小的数(java实现)

• 问题描述

几乎每一个人都用 乘法表。但是你能在乘法表中快速找到第 k 小的数字吗？

给定高度 m 、宽度 n 的一张 m * n的乘法表，以及正整数 k，你需要返回表中第 k 小的数字。

• 解决方案

首先编写一个计算乘法表中小于等于某个整数的数的个数的函数，然后运用二分查找进行搜索，代码如下：

public int findKthNumber(int m, int n, int k) {
        int low = 1;
        int high = m * n;
        while(low + 1 < high){
            int mid = low + ((high - low) >> 1);
            if(count(mid,m,n) >= k)
                high = mid;
            else
                low = mid;
        }
        if(count(low,m,n) == k)
            return low;
        else
            return high;
    }
    
    //计算乘法表中小于等于v的有多少个数
    public static int count(int v,int m,int n){
        int res = 0;
        for(int i = 1;i <= m;i ++){
            res = res + Math.min(v / i,n);
            if(v / i == 0)
                break;
        }
        return res;
    }