ARC059 E - Children and Candies（dp）

题意：

解法：

令d[i][j]表示前i个人,已经分了j个的总权值.

枚举第i+1个人分k个,d[i+1][j+k]+=d[i][j]*(a[i+1]^k+(a[i+1]+1)^k....b[i+1]^k).
其中(a[i+1]^k+(a[i+1]+1)^k....b[i+1]^k)部分可以用前缀和预处理.

最后答案为d[n][c].

算法复杂度O(n^3)

code：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=2e6+5;
const int mod=1e9+7;
int d[444][444];
int p[444][444];
int a[maxm];
int b[maxm];
int n,c;
//d[i][j]表示前i个人已经分了j个的总和
void solve(){
    cin>>n>>c;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    int ma=0;
    for(int i=1;i<=n;i++){
        cin>>b[i];
        ma=max(ma,b[i]);
    }
    for(int i=0;i<=c;i++){//i次方
        for(int j=1;j<=ma;j++){//底数j
            if(i==0)p[i][j]=1;
            else p[i][j]=p[i-1][j]*j%mod;
        }
    }
    for(int i=0;i<=c;i++){
        for(int j=1;j<=ma;j++){
            p[i][j]=(p[i][j]+p[i][j-1])%mod;
        }
    }
    for(int k=0;k<=c;k++){
        d[1][k]=(p[k][b[1]]-p[k][a[1]-1])%mod;
    }
    for(int i=1;i<n;i++){
        for(int j=0;j<=c;j++){
            if(!d[i][j])continue;
            for(int k=0;k<=c;k++){
                if(j+k>c)break;
                int temp=(p[k][b[i+1]]-p[k][a[i+1]-1])%mod;
                d[i+1][j+k]=(d[i+1][j+k]+temp*d[i][j]%mod)%mod;
            }
        }
    }
    int ans=d[n][c];
    ans=(ans%mod+mod)%mod;
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    solve();
    return 0;
}

---