前言
Weekly Contest 118 的 强整数:
给定两个非负整数 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
解题思路
本题只需要找出正整数中满足 x^i + y^j 且 <=bound 的数即可,所以只需要用两重 for 循环找出满足条件的数字即可。
注意事项:
循环中会出现重复的值,例如
输入: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
5 = 2^2 + 3^0
7 = 2^2 + 3^1
9 = 2^3 + 3^0
10 = 2^0 + 3^2
其中重复的情况为
5 = 2^1 + 3^1
5 = 2^2 + 3^0
可以选择先使用 Set 进行结果去重
执行超时的情况,例如输入:x = 1, y = 3, bound = 100。所以可以选择循环次数为 bound(x^bound>bound 恒成立),而不是 Integer.MAX_VALUE。
实现代码
/**
* 970. 强整数
* @param x
* @param y
* @param bound
* @return
*/
public List<Integer> powerfulIntegers(int x, int y, int bound) {
// 选择使用 Set 是因为结果中会出现重复值
Set<Integer> set = new HashSet<>();
// 循环次数为 bound,而不是 Integer.MAX_VALUE,防止执行超时
for (int i = 0; i < bound; i++) {
int tmp1 = (int) Math.pow(x, i);
// 如果第一个数就大于 bound,直接中断循环,防止执行超时
if (tmp1 > bound) {
break;
}
// 循环次数为 bound,而不是 Integer.MAX_VALUE,防止执行超时
for (int j = 0; j < bound; j++) {
int tmp2 = (int) (Math.pow(y, j));
int tmp = tmp1 + tmp2;
// 判断是否超出 bound
if (tmp <= bound) {
set.add(tmp);
} else {// 超出 bound 直接中断循环,因为后续的数字都会超出 bound
break;
}
}
}