Leetcode 668. 乘法表中第k小的数 C++

Leetcode 668. 乘法表中第k小的数

题目

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

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

测试样例

例 1：

输入: m = 3, n = 3, k = 5
输出: 3
解释: 
乘法表:
1	2	3
2	4	6
3	6	9

第5小的数字是 3 (1, 2, 2, 3, 3).

例 2：

输入: m = 2, n = 3, k = 6
输出: 6
解释: 
乘法表:
1	2	3
2	4	6

第6小的数字是 6 (1, 2, 2, 3, 4, 6).

注意：

1. m 和 n 的范围在 [1, 30000] 之间。
2. k 的范围在 [1, m * n] 之间。

题解

二分查找
显然，我们是从[1,m*n]区间中找到第k小的数target。我们构造一个函数，计算表中比当前数mid小的数的个数cnt，如果cnt>k，则target应该在[l,mid]；否则，则在[mid,r]上。
对于统计比当前数小的数的个数的方法为cnt += min(now/i,n)，其中i从1~m，也就是计算表中每行有多少个数比当前数小。
详细过程见代码

代码

	int countLess(int now,int m,int n){		//计算m*n乘法表中，小于等于now的数的个数
        int cnt=0;
        for(int i=1; i<=m; i++){
            cnt += min(now/i,n);
        }
        //cout<<now<<" "<<cnt<<endl;
        return cnt;
    }
    int findKthNumber(int m, int n, int k) {
        int l = 1,r = m*n,mid,cnt;
        if(k==l)    return 1;
        else if(k == r) return r;
        while(l < r){
            mid = (l+r)/2;
            cnt = countLess(mid,m,n);
            if(cnt < k)    l = mid+1;
            else    r = mid;
        }
        return l;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/kth-smallest-number-in-multiplication-table
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。