uvaoj 136 Ugly Numbers 优先队列使用

uvaoj 136 Ugly Numbers 优先队列使用

丑数是不能被除2,3,5以外的其他素数整除的数，把丑数从小到大排列，1,2,3,4,5，6,8,9,10,12,15.......，求第1500个丑数。

对于丑数x，2x，3x，5x，都是丑数，我们可以从小到大生成丑数，将生成的丑数保存起来，每次取出来最小的一个，然后生成三个丑数，但这个会重复，可以使用set去重。

代码如下：

/*************************************************************************
	> File Name: 136.cpp
	> Author: gwq
	> Mail: gwq5210@qq.com 
	> Created Time: 2015年01月21日 星期三 16时27分34秒
 ************************************************************************/

#include <cmath>
#include <ctime>
#include <cctype>
#include <climits>
#include <cstdio>
#include <cstdlib>
#include <cstring>

#include <map>
#include <set>
#include <queue>
#include <stack>
#include <string>
#include <vector>
#include <sstream>
#include <iostream>
#include <algorithm>

#define INF (INT_MAX / 10)
#define clr(arr, val) memset(arr, val, sizeof(arr))
#define pb push_back
#define sz(a) ((int)(a).size())

using namespace std;
typedef set<int> si;
typedef vector<int> vi;
typedef map<int, int> mii;
typedef long long ll;

const double esp = 1e-5;

ll mp[] = {2, 3, 5};

int main(int argc, char *argv[])
{
	priority_queue<ll, vector<ll>, greater<ll> > q;
	set<ll> s;
	q.push(1);
	s.insert(1);
	int cnt = 1;
	ll ans = 0;
	for (cnt = 1; ; ++cnt) {
		ll t = q.top();
		q.pop();
		if (cnt == 1500) {
			ans = t;
			break;
		}
		for (int i = 0; i < 3; ++i) {
			ll x = t * mp[i];
			if (!s.count(x)) {
				q.push(x);
				s.insert(x);
			}
		}
	}

	printf("The 1500'th ugly number is %lld.\n", ans);
	return 0;
}