970. 强整数
题目描述:
给定三个整数 x 、 y 和 bound ,返回 值小于或等于 bound 的所有 强整数 组成的列表 。
如果某一整数可以表示为 xi + yj ,其中整数 i >= 0 且 j >= 0,那么我们认为该整数是一个 强整数 。
你可以按 任何顺序 返回答案。在你的回答中,每个值 最多 出现一次。
示例 1:
输入:x = 2, y = 3, bound = 10
输出:[2,3,4,5,7,9,10]
解释:
2 = 20 + 30
3 = 21 + 30
4 = 20 + 31
5 = 21 + 31
7 = 22 + 31
9 = 23 + 30
10 = 20 + 32
示例 2:
输入:x = 3, y = 5, bound = 15
输出:[2,4,6,8,10,14]
解题思路
转换思路,不去判断i是否为强数,而是去构造强数.(我最开始的思路是,去判断每一个小于bound的数是否为强整数,发现总是超时)。
分情况讨论 + 哈希表:
1、若x = y = 1 ,直接返回 2
2、若x || y == 1 , 则,强数为,pow(y,i) + 1
3、若x != 1 && y != 1;强数为 pow(y,i) + pow(x,i), 如果 x > y 用 交换x、y的值,后续可以降低时间复杂度。
用数组a保存pow(x,i)。两重for()循环,第一次为pow(y,i),第二层为数组a,所有的强数为 pow(y,i) + pow(x,i) 。注意要去重,这里采用哈希表去重。
class Solution {
public:
vector<int> powerfulIntegers(int x, int y, int bound) {
vector<int>a;
vector<int>ans;
//mp用于去重
unordered_map<int,int>mp;
if(bound <= 1){
return ans;
}
//情况一 x == y == 1时
if(x == 1 && y == 1){
ans.push_back(2);
return ans;
}
//将x 设置为较小值
if(x > y){
swap(x,y);
}
//情况二
if(x == 1){
for(int j = 0 ; pow(y,j)<bound ;j++){
ans.push_back(1+pow(y,j));
}
return ans;
}
//情况三
int z = x;
a.push_back(1);
while(z < bound){
//用数组保存满足要求的 x 的幂次
a.push_back(z);
z *= x;
}
for(int i = 0 ; pow(y,i)<bound;i++){
int tp = pow(y,i);
for(auto h : a){
//哈希表去重
if(tp + h <= bound && !mp[tp + h]){
ans.push_back(tp + h);
mp[tp + h] = 1;
}
}
}
return ans;
}
};