ARC 059E Children and Candies DP

题意:N个人,C个糖果,每个人有性质xi,若第i个人分到a个糖果,则其高兴度为x[i]^a..
n个人的高兴度定义为:每个人高兴度的乘积,f(x1,x2,..xn):为所有分配方案高兴度的累加和

现在每个人的性质x[i]范围为a[i]~b[i] N,C<=400 求

设dp[i][j] i个人分j个苹果的所有方案的累加和
所有方案的累加和 分类统计 考虑第i个人分到多少苹果.
当第i个人分到m个苹果 前面i-1人分j-m个苹果的每个小方案都要乘上(a[i]^m+...b[i]^m) 化简相当于前面方案的累加和:dp[i-1][j-m]乘上(a[i]^m+...b[i]^m)
用前缀和优化下 即可达到O(N^3)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=4e2+20;
const ll mod=1e9+7;
ll n,c,a[N],b[N];
ll dp[N][N];
ll p[N][N],pre[N][N];
int main()
{
	p[0][0]=1;//i^j
	for(ll i=1;i<=400;i++)
	{
		p[i][0]=1;
		pre[i][0]=1;
		for(ll j=1;j<=400;j++)
		{
			p[i][j]=(p[i][j-1]*i)%mod;		
		}
	}
	//pre[i][j]  1^j+...i^j
	for(ll i=1;i<=400;i++)
		for(ll j=0;j<=400;j++)
			pre[i][j]=(pre[i-1][j]+p[i][j])%mod;
	while(cin>>n>>c)
	{
		for(int i=1;i<=n;i++)
			scanf("%I64d",&a[i]);
		for(int i=1;i<=n;i++)
			scanf("%I64d",&b[i]);
		memset(dp,0,sizeof(dp));
		dp[0][0]=1;
		for(int i=1;i<=n;i++)
		{
			for(int j=0;j<=c;j++)
			{
				for(int x=0;x<=j;x++)
				{
					ll t=(pre[b[i]][x]-pre[a[i]-1][x]+mod)%mod;
					dp[i][j]=(dp[i][j]+dp[i-1][j-x]*t)%mod;
				}
			}
		}
		cout<<dp[n][c]<<endl;	
	}
	return 0;
}