例题5-7 丑数（Ugly Numbers，UVa 136）

最新推荐文章于 2026-01-27 09:39:12 发布

原创最新推荐文章于 2026-01-27 09:39:12 发布 · 343 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

《算法竞赛入门经典(第2版)》专栏收录该内容

209 篇文章

订阅专栏

本文介绍了一种使用priority_queue和set数据结构来寻找第1500个丑数的方法。丑数定义为只包含质因数2、3和5的正整数。文章提供了两种实现思路的代码示例，一种利用了优先队列，另一种则直接使用集合。

原题链接：https://vjudge.net/problem/UVA-136
分类：<priority_queue>
备注：priority_queue的使用

紫书上的分析：对于任意丑数x，2x，3x，5x也都是丑数。

即使通过运算得到了1500个数，第1500个数也可能不是正好的第1500个丑数，可能是更大的数。所以要从小往大数，这样应用priority_queue就需要greater。

(作者的priority_queue(经改动))代码如下：

#include<cstdio>
#include<vector>
#include<queue>
#include<set>
using namespace std;
typedef long long ll;
int main(void)
{
	priority_queue<ll, vector<ll>, greater<ll> >pq;//小数优先
	set<ll>s;
	pq.push(1);
	s.insert(1);
	for (int i = 1;; i++)
	{
		ll x = pq.top(); pq.pop();
		if (i == 1500)
		{
			printf("The 1500'th ugly number is %lld.\n", x);
			break;
		}
		ll x1 = 2 * x, x2 = 3 * x, x3 = 5 * x;
		if (!s.count(x1)) s.insert(x1), pq.push(x1); 
		if (!s.count(x2)) s.insert(x2), pq.push(x2);
		if (!s.count(x3)) s.insert(x3), pq.push(x3);
	}
	return 0;
}

这里是我自己的方法:

#include<cstdio>
#include<set>
using namespace std;
int main(void)
{
	set<long long>s;
	int cnt = 0;
	s.insert(1);
	for (set<long long>::iterator it = s.begin();; ++it)
	{
		cnt++;
		if (cnt == 1500) { printf("The 1500'th ugly number is %lld.\n", *it); break; }
		s.insert(*it * 2);
		s.insert(*it * 3);
		s.insert(*it * 5);
	}
	return 0;
}