【codevs1246】丑数，STL与取模大质数的好处

丑数 USACO
时间限制: 1 s
空间限制: 128000 KB
题目等级 : 钻石 Diamond
题解
题目描述 Description
对于一给定的素数集合 S = {p1, p2, …, pK},
来考虑那些质因数全部属于S 的数的集合。这个集合包括，p1, p1p2, p1p1, 和 p1p2p3 (还有其它)。这是个对于一个输入的S的丑数集合。
注意：我们不认为1 是一个丑数。
你的工作是对于输入的集合S去寻找集合中的第N个丑数。longint(signed 32-bit)对于程序是足够的。

输入描述 Input Description
第 1 行: 二个被空间分开的整数:K 和 N ， 1<= K<=100 ， 1<= N<=100,000.
第 2 行: K 个被空间分开的整数:集合S的元素

输出描述 Output Description
单独的一行，写上对于输入的S的第N个丑数。

样例输入 Sample Input
4 19
2 3 5 7

样例输出 Sample Output
27
写在前面：mod大法好= =
——————————————————————————————————————————————
思路：正确作法是贪心，不过我打了个十分暴力的STL堆＋map，每次从堆中取出最小一个数x，然后再向堆里加入x*a[i]（用map判重），但是这样最后一个点是过不了的，于是狠下心来放弃map，尝试用了一下取模大质数判重，没想到就过了……

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue> 
#include<map>
#define mod 40000003
using namespace std;
int n,k,a[110];
long long ans;
priority_queue <long long,vector<long long>,greater<long long> > team;
bool flag[40000004];
main()
{
    scanf("%d%d",&k,&n);
    for (int i=1;i<=k;i++) scanf("%d",&a[i]),team.push(a[i]);
    for (int i=1;i<=n-1;i++)
    {
        ans=team.top();
        team.pop();
        for (int j=1;j<=k;j++)
        if (!flag[ans*a[j]%mod]) flag[ans*a[j]%mod]=true,team.push(ans*a[j]);
    }
    printf("%d",team.top());
}