970. 强整数(Python)

作者: 玖月晴 | 来源:发表于2019-06-01 18:02 被阅读0次

    更多精彩内容,请关注【力扣简单题】

    题目

    难度:★★☆☆☆
    类型:数学,排列组合

    给定两个正整数 x 和 y,如果某一整数等于 x^i + y^j,其中整数 i >= 0 且 j >= 0,那么我们认为该整数是一个强整数。

    返回值小于或等于 bound 的所有强整数组成的列表。

    你可以按任何顺序返回答案。在你的回答中,每个值最多出现一次。

    提示
    1 <= x <= 100
    1 <= y <= 100
    0 <= bound <= 10^6

    示例

    示例 1
    输入:x = 2, y = 3, bound = 10
    输出:[2,3,4,5,7,9,10]
    解释:
    2 = 2^0 + 3^0
    3 = 2^1 + 3^0
    4 = 2^0 + 3^1
    5 = 2^1 + 3^1
    7 = 2^2 + 3^1
    9 = 2^3 + 3^0
    10 = 2^0 + 3^2

    示例 2
    输入:x = 3, y = 5, bound = 15
    输出:[2,4,6,8,10,14]

    解答

    这道题的难点在于如何进行i和j的搜索。我们可以使用类似穷举法的方式,找到所有可能的(i, j)组合,注意这里的i和j有可能相等,并且从零开始,上限需要我们寻找。

    如何寻找上限?我们首先需要设置(i, j)搜索范围,这与输入的两个数字x和y有关,为了寻找尽可能全面,我们需要选出以边界bound为底,分别以x和y为幂的对数log(bound, x)和log(bound, y)中的较大值,并且取整,因此搜索范围初步定为从0到int(log(bound, min(x, y)))(含)。

    举个栗子,如果输入x=2,y=3,bound=10,搜索空间的上限是int(min(log(10, 2), log(10, 3)))+1 = int(log(10, min(2, 3)))+1=int(log(10, 2))+1 = int(3.32...) = 3,i和j的搜索范围是[0, 1, 2, 3],在python中表示为list(range(4))。(注意这里是左闭右开的区间)

    python中product(list, repeat=2)可以返回列表自身与自身元素组成的笛卡尔积元组,上述(i, j)元素的搜索范围可以表示为product(range(4), repeat=2),这时会生成一个迭代器,迭代会获得下面的(i, j)元组:

    0 0
    0 1
    0 2
    0 3
    1 0
    1 1
    1 2
    1 3
    2 0
    2 1
    2 2
    2 3
    3 0
    3 1
    3 2
    3 3
    

    这些i,j囊括了不大于bound的所有pow(x, i)+pow(y, i)可能,我们要做的从其中选择出能使pow(x, i)+pow(y, i)不大于bound的所有可能。

    这里有一个问题需要注意,就是log(b, a)(计算以b为底a的对数)函数的输入参数存在合法范围,a不能小于1,b不能小于2,因此我们使用max()函数对输入进行下限的限制:log(max(b, 1), max(a))。

    from itertools import product
    from math import log
    
    
    class Solution:
        def powerfulIntegers(self, x, y, bound):
            """
            :param x: int
            :param y: int
            :param bound: int
            :return: List[int]
            """
            res = []
            rg = int(log(max(bound, 1), min(max(x, 2), max(y, 2))))+1
            for i, j in product(range(rg), repeat=2):
                num = x ** i + y ** j
                if num <= bound:
                    res.append(num)
            return list(set(res))
    

    如有疑问或建议,欢迎评论区留言~

    相关文章

      网友评论

        本文标题:970. 强整数(Python)

        本文链接:https://www.haomeiwen.com/subject/aqpatctx.html