给定两个正整数 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))