来源:力扣(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
|