LeetCode-Python-970. 强整数

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

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

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

 

示例 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]

 

提示：

• 1 <= x <= 100
• 1 <= y <= 100
• 0 <= bound <= 10^6

思路：

动态规划，用dpx和dpy两个数组分别存小于bound的x和y的可能的幂倍数，然后双重循环组合即可。

注意当x 或者y 为 1 的时候，需要手动设定dpx， dpy。

class Solution(object):
    def powerfulIntegers(self, x, y, bound):
        """
        :type x: int
        :type y: int
        :type bound: int
        :rtype: List[int]
        """
        dpx = list()
        dpy = list()
        res = list()
        if x != 1:
            t = 0
            while( x **t < bound):
                dpx.append(x **t)
                t += 1
        else:
            dpx = [1]
        if y != 1:

            t = 0
            while(y **t < bound):
                dpy.append(y ** t)
                t += 1
        else:
            dpy = [1]

        for i, x in enumerate(dpx):
            for j, k in enumerate(dpy):
                t = x + k
                if t <= bound:
                    res.append(t)
        return list(set(res))