题意:问第n(n <= 1500)小的丑数(质因数只有2或3或5)是几。
题目链接:http://poj.org/problem?id=1338
——>>1, 2, 3, 4, 5, 6, 8, ...
假设小根堆存以上丑数,那么每次取出最小的数,这个最小的数nMin,它可以生成三个数:nMin * 2, nMin * 3, nMin * 5,将这三个数放入小根堆继续,一直复筛出1500个丑数为止。
小根堆可用优先队列来替代。
#include <cstdio>
#include <queue>
#include <vector>
using std::priority_queue;
using std::vector;
using std::greater;
const int MAXN = 1500;
long long nRet[MAXN];
void GetUglyNumbers()
{
int nCnt = 0;
priority_queue<long long, vector<long long>, greater<long long> > pq;
pq.push(1);
while (nCnt < MAXN)
{
long long nMin = pq.top();
pq.pop();
if (nRet[nCnt] != nMin)
{
nRet[++nCnt] = nMin;
pq.push(nMin * 2);
pq.push(nMin * 3);
pq.push(nMin * 5);
}
}
}
int main()
{
int n;
GetUglyNumbers();
while (scanf("%d", &n) == 1 && n)
{
printf("%I64d\n", nRet[n]);
}
return 0;
}