AC日记——丑数 codevs 1246

1246 丑数

 

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

数据范围及提示 Data Size & Hint

 

 

思路：

　　刚开始想用堆做结果最后一个点TLE

　　现在看了别人的题解才明白还有动态规划

　　认定1是第一个丑数（虽然它不是）

　　然后我们从1开始往后推

　　用一个now数组来记录k个质数的当前对应的dis下标

　　然后我们每次更新一个推出的属于s的数

　　这个s一定是当前最小的

　　因为它由k个dis取出的最小

　　好吧，我也说不清，看代码

 

来，上代码：

#include <cstdio>

#define INF 0x7fffffff

using namespace std;

int n,k,ai[101],now[101];

long long int dis[100001];

int main()
{
    scanf("%d%d",&k,&n);
    for(int i=1;i<=k;i++) scanf("%d",&ai[i]),now[i]=1;
    dis[1]=1;
    for(int i=2;i<=n+1;i++)
    {
        dis[i]=INF;
        for(int j=1;j<=k;j++)
        {
            while(ai[j]*dis[now[j]]<=dis[i-1]) now[j]++;
            if(ai[j]*dis[now[j]]<dis[i]) dis[i]=ai[j]*dis[now[j]];
        }
    }
    printf("%lld\n",dis[n+1]);
    return 0;
}