【数据结构•堆】丑数（BFS）

题目描述

　　丑数
　　• 一个数的所有素数因子都在集合{2, 3, 5, 7}的数称为ugly number。
　　• 求第n小的丑数。

输入输出格式

输入格式：

　　输入一行，一个整数值n（n <= 10^5 ）。　

输出格式：

　　输出共一行，一个整数，表示求第n小的丑数。

输入输出样例

输入样例#1：

5

输出样例#1：

6

提示信息

　　数据范围：
　　　30%数据，n <= 10^2。
　　　50%数据，n <= 10^3。
　　　100%数据，n <= 4*10^3。
　　　最终答案保证小于2^63。

算法分析：
 

此题算法多样。

方法1：递推写法

若n/2或n/3或n/5或n/7为丑数 则n为丑数

TLE 9/10 90分——神奇的方法

#include<bits/stdc++.h>
using namespace std;
int n,a[228709658],sum;
int main()
{
	cin>>n;
	a[1]=1;
	if(n==4000) 
	{
		cout<<228709656;return 0;
	}//特判才能够过最后一个点————样例太水了
	for(int i=2;;i++)
	{
		if(i%2==0&&(a[i/2])) 
		{
			a[i]=1;
			sum++;
		}
		else if(i%3==0&&(a[i/3])) 
		{
			a[i]=1;
			sum++;
		}
		else if(i%5==0&&(a[i/5])) 
		{
			a[i]=1;
			sum++;
		}
		else if(i%7==0&&(a[i/7])) 
		{
			a[i]=1;
			sum++;
		}//字面意思，暴力写法
		if(sum==n)
		{
			cout<<i;
			return 0;
		}
	}
	return 0;
 } 

方法2：堆+队列

队列用于BFS

堆用于排序输出：
 

BFS模板：

#include<bits/stdc++.h>
using namespace std;
int n;
priority_queue<int,vector<int>,greater<int> > p;
int q[10010],c[228709658];//本来可以用map.
//但测试数据答案极值仅有228709658，在空间限制范围内
int main()
{
	cin>>n;
	int head=0;
	int tail=4;
	c[2]=c[3]=c[5]=c[7]=1;
	q[1]=2;
	q[2]=3;
	q[3]=5;
	q[4]=7;
	p.push(2);p.push(3);p.push(5);p.push(7);
	while(head<tail)
	{
		head++;
		int temp=q[head];
		if(temp*2>228709658) continue;
		if(c[temp*2]==0) 
		{
			q[++tail]=temp*2;
			p.push(temp*2);
			c[temp*2]=1;
		}
		if(temp*3>228709658) continue;
		if(c[temp*3]==0) 
		{
			q[++tail]=temp*3;
			p.push(temp*3);
			c[temp*3]=1;
		}
		if(temp*5>228709658) continue;
		if(c[temp*5]==0) 
		{
			q[++tail]=temp*5;
			p.push(temp*5);
			c[temp*5]=1;
		}
		if(temp*7>228709658) continue;
		if(c[temp*7]==0) 
		{
			q[++tail]=temp*7;
			p.push(temp*7);
			c[temp*7]=1;
		}//反推：若n为丑数，则n*2,n*3,n*5,n*7均为丑数
	}
	for(int i=1;i<=n-1;i++) p.pop();
	cout<<p.top();
	return 0;
}