首页 > 代码库 > poj - 1338 - Ugly Numbers(优先队列)
poj - 1338 - Ugly Numbers(优先队列)
题意:问第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; }
poj - 1338 - Ugly Numbers(优先队列)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。