CF543A

Codeforces 302 Div.1 A

题意是说, 给1<=n,m,b<=500, 1<=mod<=1000000007, 以及n个数0<=a[i]<=500, 求满足v[1]+...+v[n]=m, a[1]*v[1]+...+a[n]*v[n]<=b的v的组数(取模mod)的结果.

思路自然是dp. 但是直接dp会超时, 需要剪枝. 具体思路直接见代码.

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<vector>
#include<map>
#include<math.h>
using namespace std;

typedef long long ll;

int n,m,b,a[505]={0};
ll mod;
ll dp[505][505]={0},c[1005][1005]={0};

int main()
{
    int i,j,k,l;
    scanf("%d %d %d %I64d",&n,&m,&b,&mod);
    for(i=0;i<n;i++)
    {
        int x;
        scanf("%d",&x);
        a[x]++;
    }
    for(i=0;i<1005;i++) c[i][i]=c[i][0]=1%mod;
    for(i=2;i<1005;i++) for(j=1;j<i;j++) c[i][j]=(c[i-1][j-1]+c[i-1][j])%mod;
    dp[0][0]=1;
    for(i=0;i<505;i++)
    {
        //cout<<i<<endl;
        if(a[i]==0) continue;
        for(j=m-1;j>=0;j--)
        {
            for(k=b-(m-j)*i;k>=0;k--)
            {
                if(dp[j][k]==0) continue;
                for(l=1;j+l<=m&&k+l*i<=b;l++)
                {
                    dp[j+l][k+l*i]+=dp[j][k]*c[l+a[i]-1][a[i]-1]%mod;
                    dp[j+l][k+l*i]%=mod;
                }
            }
        }
    }
    ll r=0;
    for(i=0;i<=b;i++) r+=dp[m][i],r%=mod;
    printf("%I64d\n",r);
    return 0;
}