USACO-Section3.1 Humble Numbers【小顶堆/priority_queue】

题目描述：

对于一给定的素数集合 S = {p1, p2, …, pK},考虑一个正整数集合，该集合中任一元素的质因数全部属于S。这个正整数集合包括，p1、p1*p2、p1*p1、p1*p2*p3…(还有其它)。该集合被称为S集合的“丑数集合”。
注意：我们不认为1 是一个丑数。
你的工作是对于输入的集合S去寻找“丑数集合”中的第N个“丑数”。所有答案可以用longint(32位整数)存储。
补充：丑数集合中每个数从小到大排列，每个丑数都是素数集合中的数的乘积，第N个“丑数”就是在能由素数集合中的数相乘得来的（包括它本身）第n小的数。

INPUT FORMAT:

第 1 行: 二个被空格分开的整数:K 和 N ， 1<= K<=100 ， 1<= N<=100,000.
第 2 行: K 个被空格分开的整数:集合S的元素

OUTPUT FORMAT:

单独的一行，输出对于输入的S的第N个丑数。

---

SAMPLE INPUT

4 19
2 3 5 7

---

SAMPLE OUTPUT

27

---

解题思路：

这道题的第一想法就是用优先队列解决，将队列的头（即最小值）p出队列，p的值乘以集合中的每一个数压入队列。当出队列的个数为N时返回该值。想法很美好，但是遇到了好几个问题。

第一个问题是数据重复，由于priority_queue没有find函数，所以我们只能避免输入重复数据，我用的方法是{2，3，5，7}这个队列，从7到2，如果p可以7整除，那么它只会压入p*7，如果p只可以被2整除，则它会压入所有的值。

第二个问题是内存不够。emmmmm，这个就比较尴尬，有的人可能会想为什么不在size为N的时候就返回最大值，但是我用的这个方法没法保证每次压入的都是下一个值，所以我用了一个小优化，记录队列中的最大值，如果这时size大于等于N了，那么这个最大值肯定是大于结果的，那么只要要压入的值大于N，就跳过，可以极大的节省内存。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;

long long int K,N;
long long int a[110];
long long int ccount=0;//记录目前出队列的个数，当为N时是答案 
long long int allmax=0;//记录目前队列的最大值，一个小优化，用于节省空间，当优先队列size大于N时，只有比目前最大值小的数才能压入队列。
priority_queue<long long int,vector<long long int>,greater<long long int> > que;

long long int judge(){ 
    while(1){
        long long int p=que.top();que.pop();
        ccount++;
        if(ccount==N)return p;
        int flag=0;
        for(int i=K-1;i>=0;i--){//由于priority_queue没有find()函数，只好避免录入重复值 
            if(p%a[i]==0){
                flag=i;
                break;
            }
            else continue;
        }
        for(int i=flag;i<K;i++){
                if(que.size()>=N){
                    if(a[i]*p>allmax)break;
                }
                que.push(a[i]*p);
                allmax=max(a[i]*p,allmax);
        }       
    }
} 
int main(){
    FILE *fin  = fopen ("humble.in", "r");
    FILE *fout = fopen ("humble.out", "w");
    fscanf(fin,"%d%d",&K,&N);
    for(int i=0;i<K;i++){
        fscanf(fin,"%lld",&a[i]);
        que.push(a[i]);
    }
    allmax=a[K-1];
    fprintf(fout,"%lld\n",judge());
    exit(0);
}