hdu 2110 Crisis of HDU(母函数)

题意:http://acm.hdu.edu.cn/showproblem.php?pid=2110

组合问题,需要用到母函数,但是x的指数是由题意决定的,而且是不断变化的。这就需要灵活的使用母函数,而且涉及到取模,不要忘了。

#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int c1[10005],c2[10005];
int main()
{
    //freopen("cin.txt","r",stdin);
    int n,sum,aim,num;
    int p[101],m[101];
    while(cin>>n&&n){
        int i,j,k;
        sum=0;
        memset(c1,0,sizeof(c1));
        memset(c2,0,sizeof(c2));
        for(i=0;i<n;i++){
            scanf("%d%d",&p[i],&m[i]);
            sum+=p[i]*m[i];
        }
        if(sum%3==0){
            aim=sum/3;  //财产分割的数目。
            num=p[0]*m[0];
            for(i=0;i<=num;i+=p[0])c1[i]=1;
            for(i=1;i<n;i++){
                for(j=0;j<=p[i]*m[i];j+=p[i]){
                    for(k=0;k<=num;k++){
                        if(j+k>10000)break;
                        c2[j+k]+=c1[k];
                        c2[j+k]%=10000;  //每步都模运算
                    }
                }
                num+=p[i]*m[i];
                for(j=0;j<=num;j++){
                    c1[j]=c2[j];
                    c2[j]=0;
                }
            }
        }
        if(c1[aim])printf("%d\n",c1[aim]%10000);
        else printf("sorry\n");
    }
    return 0;
}


你可能感兴趣的:(HDU)