POJ 3046

最新推荐文章于 2020-04-05 14:08:43 发布

原创最新推荐文章于 2020-04-05 14:08:43 发布 · 2k 阅读

1 ·

CC 4.0 BY-SA版权

ACM 动态规划专栏收录该内容

17 篇文章

订阅专栏

题意：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;
}