首页 > 代码库 > HDU 2189 悼念512汶川大地震遇难同胞――来生一起走 --生成函数

HDU 2189 悼念512汶川大地震遇难同胞――来生一起走 --生成函数

这题跟上两题也差不多。

把150以内的素数找出来,把素数的值看做硬币的面值,每个硬币的个数即ceil(150/prime[i]),因为再多也没用,最多组成n=150就行了,所以又回到了找硬币问题。用生成函数解之。

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define N 207

int prime[N],flag[N],num[N];
int c[7002],tc[7002];
int ci;

void doit()
{
    int i,j,n;
    ci = 0;
    for(i=2;i<=150;i++)
        flag[i]=1;
    for(i=2;i<=150;i++)
    {
        if(flag[i])
           prime[ci++]=i;
        for(j=0;j<ci&&i*prime[j]<=150;j++)
        {
            flag[i*prime[j]]=0;
            if(i%prime[j]==0)
               break;
        }
    }
}

int main()
{
    doit();
    int i,j,k,t,n;
    int maxi = 0;
    for(i=0;i<ci;i++)
    {
        num[i] = 150/prime[i]+1;
        maxi += prime[i]*num[i];
    }
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(i=0;i<=maxi;i++)
            c[i] = tc[i] = 0;
        for(i=0;i<=prime[0]*num[0];i+=prime[0])
            c[i] = 1;
        for(i=1;i<ci;i++)
        {
            for(j=0;j<=maxi;j++)
            {
                for(k=0;k+j<=maxi && k<=prime[i]*num[i];k+=prime[i])
                    tc[k+j] += c[j];
            }
            for(j=0;j<=maxi;j++)
            {
                c[j] = tc[j];
                tc[j] = 0;
            }
        }
        printf("%d\n",c[n]);
    }
    return 0;
}
View Code