设计算法，找到质因数只有3，5或7的第k个数。

设计算法，找到质因数只有3，5或7的第k个数。  

2014-06-18 14:02:27|  分类： 默认分类|举报|字号 订阅

我们可以用3个队列来维护这些数。第1个队列负责乘以3，第2个队列负责乘以5， 第3个队列负责乘以7。算法描述如下：

1. 初始化结果res=1和队列q3,q5,q7
2. 分别往q3,q5,q7插入1*3,1*5,1*7
3. 求出三个队列的队头元素中最小的那个x，更新结果res=x
4. 如果x在：
    q3中，那么从q3中移除x，并向q3，q5，q7插入3*x,5*x,7*x
    q5中，那么从q5中移除x，并向q5，q7插入5*x,7*x
    q7中，那么从q7中移除x，并向q7插入7*x
5. 重复步骤3－5，直到找到第k个满足条件的数

注意，当x出现在q5中，我们没往q3中插入3*x，那是因为这个数在q5中已经插入过了。

代码如下：

#include <iostream>
#include <queue>
using namespace std;

int mini(int a, int b){
    return a < b ? a : b;
}
int mini(int a, int b, int c){
    return mini(mini(a, b), c);
}
int get_num(int k){
    if(k <= 0) return 0;
    int res = 1, cnt = 1;
    queue<int> q3, q5, q7;
    q3.push(3); q5.push(5); q7.push(7);
    for(; cnt<k; ++cnt){
        int v3 = q3.front();
        int v5 = q5.front();
        int v7 = q7.front();
        res = mini(v3, v5, v7);
        if(res == v7){
            q7.pop();
        }
        else{
            if(res == v5){
                q5.pop();
            }
            else{
                if(res == v3){
                    q3.pop();
                    q3.push(3*res);
                }
            }
            q5.push(5*res);
        }
        q7.push(7*res);
    }
    return res;
}
int main(){
    for(int i=1; i<20; ++i)
        cout<<get_num(i)<<endl;
    return 0;
}