【leetcode】668. Kth Smallest Number in Multiplication Table

最新推荐文章于 2022-05-18 18:26:45 发布

weixin_34174322

最新推荐文章于 2022-05-18 18:26:45 发布

阅读量160

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/seyjs/p/9604157.html

本文介绍了一种类似于LeetCode 719题的解法，用于在m*n的矩阵中找到第K小的乘积。通过调整算法，将原本的减法变为乘法，并考虑了i*j与j*i的不同，利用二分查找高效解决问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目如下：

解题思路：几乎和【leetcode】719. Find K-th Smallest Pair Distance 的方法一样。只不过一个是减法一个是乘法，还有一点区别是【leetcode】719. Find K-th Smallest Pair Distance中i-j和j-i只算一个元素，而本题中i*j与j*i算两个元素。

代码如下：

class Solution(object):
    def findKthNumber(self, m, n, k):
        """
        :type m: int
        :type n: int
        :type k: int
        :rtype: int
        """
        low,high = 1, m*n

        while low <= high:
            mid = (low + high) / 2
            less,equal = 0,0
            for i in range(1,min(mid+1,m+1)):
                less += min(mid/i,n)
                if mid/i <= n and mid % i == 0:
                    equal += 1
                    less -= 1

            if less >= k:
                high = mid - 1
            elif less + equal < k:
                low = mid + 1
            elif less == k and equal == 0:
                high = mid - 1
            else:
                break
        return mid