力扣算法_668 乘法表中第k小的数

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).

思路

方法一：遍历m、n，分别计算乘积并记入列表中，最终找出第k小的数字。（超出时间限制）

方法二：找第k个小的数字，换句话说，计算某个数前面有几个数；因此可以使用二分法取数，并计算该数前面有几个数字，直到满足条件。

代码

方法一：

class Solution:
    def findKthNumber(self, m: int, n: int, k: int) -> int:
        lst = []
        for i in range(1, m+1):
            for j in range(1, n+1):
                lst.append(i*j)
        lst.sort()
        return lst[k-1]

方法二：

class Solution:
    def count(self, x, m, n):
        res, i, j = 0, m, 1
        while i >= 1 and j <= n:
            if i * j <= x:
                res += i
                j += 1
            else: i -= 1
        return res

    def findKthNumber(self, m: int, n: int, k: int) -> int:
        l, r = 1, m * n
        while l < r:
            x = (l + r) // 2
            if self.count(x, m, n) < k: l = x + 1
            else: r = x
        return l