c++ 醜數

標籤：using   整數   main   醜數   name   迴圈   include   push   排序   

/*
定義:醜數是指不能被2,3,5以外的其他素數整除的數.
問題:把醜數從小到大排列起來,結果如下 1 2 3 5 6 8 9 10 12 15 求第1500個醜數
解題思路:醜數的2,3,5倍仍然是醜數(排除重複的)
*/
#include<iostream>
#include<vector>
#include<set>
#include<queue>
using namespace std;
typedef long long LL;//定義long long 類型的別名
const int coeff[3]={2,3,5};//定義基礎數組
int main(){
    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();//第一元素出隊列 所以 第1500個醜數就是pop1499後的那個元素
        if(i==1500){
            cout<<"The 1500‘th ugly number is "<<x<<".\n";//輸出第1500個醜數
            break;//跳出迴圈
        }
        for(int j=0;j<3;j++){
            LL x2=x*coeff[j];//x2是2 3 5的倍數
            //判斷集合中有沒有x2 若沒有進入if中
            if(!s.count(x2)){
                s.insert(x2);//插入到s集合中
                pq.push(x2);//進隊列
            }
        }
    }
    return 0;
}

c++ 醜數