来源:力扣(LeetCode) 题目链接
几乎每一个人都用 乘法表。但是你能在乘法表中快速找到第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).
注意:
m 和 n 的范围在 [1, 30000] 之间。
k 的范围在 [1, m * n] 之间。
class Solution {
public:
bool check(int m,int n,int mid,int k)
{
int cnt = 0;
//这里的计数:小于等于mid的数
for(int i = 1; i <= m; i++)
cnt += min(mid / i, n);
//如果包括mid在内的cnt >= k,表明mid可以是结果,所以r = mid
//如果包括mid在内的cnt < k,表明mid一定不是结果,所以l = mid + 1
return cnt >= k;
}
int findKthNumber(int m, int n, int k) {
int l = 1,r = m * n;
//使得m为小值,check的循环会少一点
if(m > n) swap(m,n);
while(l < r)
{
//这里l+r不会超出范围,所以直接用的l + r >> 1
int mid = l + r >> 1;
if(check(m,n,mid,k))r = mid;
else l = mid + 1;
}
return r;
}
};
//这道题不是特别难,二分比较容易想到
//重点在于边界的问题,二分题的边界是一个比较麻烦的事情
//重点在cnt >= k这里的返回,和cnt统计的数包不包括mid



