有些数的素因子只有 3,5,7,请设计一个算法找出第 k 个数。注意,不是必须有这些素因子,而是必须不包含其他的素因子。例如,前几个数按顺序应该是 1,3,5,7,9,15,21。
示例 1:
输入: k = 5
输出: 9
题意有点不好理解,就是找出一个序列,该序列的素因子只有3 5 7这三个数,即为3 5 7的倍数,但是我还是不知道为什么这里面会有1。
class Solution {
public:
typedef long long LL;
int getKthMagicNumber(int k) {
unordered_map< LL , LL> ump ;
priority_queue< LL , vector<LL> , greater<LL> > pque ;
pque.push(1l);
ump[1]=1;
LL init[3] = {3,5,7};
LL ans ;
for(int i = 0 ; i < k ; i ++)
{
ans = pque.top();
pque.pop();
for(auto tmp : init)
{
LL x = ans*tmp;
if(!ump[x]) pque.push(x),ump[x]=1;
}
}
return ans;
}
};