第k个数

题目描述

有一些数的素因子只有3、5、7，请设计一个算法，找出其中的第k个数。

给定一个数int k，请返回第k个数。保证k小于等于100。

测试样例：

3

返回：7

 

之前做过一道求素因子只有2，3，5的，这类题都是用这个思想做的。

 

class KthNumber {
public:
    int findKth(int k) {
        // write code here
        vector<int> v(k + 1, 0);
        v[0] = 1;
        int cnt3 = 0;
        int cnt5 = 0;
        int cnt7 = 0;
        for(int i = 1; i <= k; ++i)
        {
            int x = min(3 * v[cnt3], min(5 * v[cnt5], 7 * v[cnt7]));
            v[i] = x;
            if(x == 3 * v[cnt3])
            {
                cnt3++;
            }
            if(x == 5 * v[cnt5])
            {
                cnt5++;
            }
            if(x == 7 * v[cnt7])
            {
                cnt7++;
            }
        }
        return v[k];
    }
};