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]

 

解题思路

打表以空间换时间，先设置两个数组分别存放x的指数的值和y的指数的值。然后两层for循环中判断两数组元素和是否小于等于bound，如果小于等于则放入set，否则跳出内层循环；然后将set内的元素依次放入vector

int pre_produce(int sub[],int a){
        int sum = 1,num = 0;
        sub[0] = 1;
        if(a == 0 || a == 1) return 0;
        while(sum <= 1e6){
            sum *= a;
            sub[++num] = sum;
        }
        return num;
    }
    vector<int> powerfulIntegers(int x, int y, int bound) {
        int subx[101] = {0},suby[101] = {0};
        vector<int> vect;
        set<int> st;
        set<int>::iterator it;
        int xi = pre_produce(subx,x);
        int yi = pre_produce(suby,y);
        for(int i=0;i<=xi;i++){
            for(int j=0;j<=yi;j++){
                if(subx[i]+suby[j] <= bound){
                    st.insert(subx[i]+suby[j]);
                }else break;
            }
        }
        for(it = st.begin();it!=st.end();it++) vect.push_back(*it);
        return vect;
    }