丑数定义:只包含2、3、5质因子的整数称为丑数ugly number
习惯上将1当作第一个丑数,前20个丑数为:1, 2, 3, 4, 5, 6, 8, 9, 10, 12, 15, 16, 18, 20, 24, 25, 27, 30, 32, 36。
题目需求:输入n ,输出第n大的丑数
解题思路:优先队列 第n个丑数必为前n-1个丑数乘上2、3、5得到的乘积的最小值,用优先队列维护一个升序的数列,不断的pop()队首元素并将其与2、3、5相乘的结果入队
每次pop() 当前队列里最小的元素出队
因此pop()操作执行n-1次后队首元素即为第n大的丑数
#include<iostream>
#include<queue>
#include<map>
using namespace std;
#define int long long
map<int,int>mp;
signed main()
{
priority_queue<int,vector<int>,greater<int> >q;//声明优先队列q 按照从小到大自动排序
int n;
cin>>n;
q.push(1);
for(int i=0;i<n-1;i++){
int tmp=q.top();
q.pop();//pop n-1次以后队列的top()就是第n大的丑数
if(mp[tmp*2]==0){
mp[tmp*2]=1;
q.push(tmp*2);
}
if(mp[tmp*3]==0){
mp[tmp*3]=1;
q.push(tmp*3);
}
if(mp[tmp*5]==0){
mp[tmp*5]=1;
q.push(tmp*5);
}
}
int ans=q.top();
printf("The %d'th ugly number is <%d>.",n,ans);
// cout<<q.top()<<endl;
}