LeetCode—— 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/powerful-integers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：循环扫描，使用set去重复，然后把set的结果存入vector ， 注意y==1，x==1，一次计算则退出。

class Solution {
public:
    vector<int> powerfulIntegers(int x, int y, int bound) {
        set<int> res;
        int result = 0;
        for(int i = 0; pow(x, i) <= bound; ++i )
        {
            for(int j = 0; pow(y, j) <= bound; ++j)
            {
                result = pow(x, i) + pow(y, j);
                if(result <= bound)
                {
                    res.insert(result);
                }
                if(y == 1)  break;    // y=1时，只需要计算1次
            }
            if(x == 1)  break; // x=1时，只需要计算1次。
        }
        set<int>::iterator it = res.begin();
        vector<int> ret;
        while(it != res.end())
        {
            ret.push_back(*it);
            ++it;
        }
        return ret;
    }
};