POJ 3046

来源:互联网 发布:淘宝网开店要钱吗 编辑:程序博客网 时间:2024/06/05 16:06

题意:T种蚂蚁,共计A只,种内没有区别,问含有x只蚂蚁的子集有多少种(x∈[S,B])

题解:动态规划,dp[i][j]代表用前i种蚂蚁凑成j只蚂蚁的集合的方法数,由于T*A太大,用滚动数组实现,最后输出的是答案mod 1000000

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;int dp[2][100005],num[1005];const int mod=1000000;int main(){    int t,a,b,s,sum,ans;    while(scanf("%d%d%d%d",&t,&a,&s,&b)!=EOF)    {        sum=0;        memset(num,0,sizeof(num));        for(int tp,i=0;i<a;i++)            scanf("%d",&tp),num[tp]++;        memset(dp,0,sizeof(dp));        dp[0][0]=1;        for(int i=1;i<=t;i++)        {            int nu=num[i],y=i&1,x=y^1;            memcpy(dp[y],dp[x],sizeof(dp[x]));            for(int j=0;j<=sum&&j<=b;j++)                for(int k=1;k<=nu;k++)                    dp[y][j+k]+=dp[x][j],dp[y][j+k]%=mod;            sum+=nu;        }        ans=0;        t=t&1;        for(int i=s;i<=b;i++)            ans+=dp[t][i],ans%=mod;        printf("%d\n",ans);    }    return 0;}