HDU 5396 Expression (数学期望+区间DP)

时间:2022-02-12 14:59:28

题目链接:

http://acm.hdu.edu.cn/showproblem.php?pid=5396


题意:

给出一个有n个数字,运算符只有"+","-","*"的表达式,每次合并相邻两项,求所有合并方式所得到的最终结果之和对1e9+7取模的值。


分析:

如果将这个过程随机化,即每次等概率地选取相邻两项合并,

记dp[i][j]为随机合并第i个到第j个数字这一段的表达式之后结果的期望,

根据期望的线性可加性,状态转移方程为

dp[i][j]=(sigma_(k=i~j-1)(dp[i][k]?dp[k+1][j]))/(j-i),

其中"?"表示第k个数与第k+1个数之间的运算符,

那么dp[1][n]即为随机合并整个表达式之后结果的期望,

乘上方案数(n-1)!即为所求的总和,

由于是取模意义下的运算,转移方程中的除法要用逆元代替,

复杂度O(n^3)。


代码:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int MAXN=105;
const ll Mod=1000000007LL;
ll a[MAXN],f[MAXN];
char s[MAXN];
ll dp[MAXN][MAXN];
void build()
{
f[0]=1LL;
for(int i=1;i<=100;i++)
f[i]=i*f[i-1]%Mod;
}
ll fp(ll a,ll k)
{
ll res=1LL;
while(k>0)
{
if(k&1)res=res*a%Mod;
a=a*a%Mod;
k>>=1;
}
return res;
}
int main()
{
build();
int n;
while(scanf("%d",&n)!=EOF)
{
for(int i=1;i<=n;i++)
scanf("%I64d",&a[i]);
scanf("%s",s+1);
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
dp[i][i]=a[i];
for(int len=1;len<n;len++)
{
for(int i=1;i+len<=n;i++)
{
int j=i+len;
for(int k=i;k<j;k++)
{
if(s[k]=='+')
dp[i][j]=(dp[i][j]+dp[i][k]+dp[k+1][j])%Mod;
else if(s[k]=='-')
dp[i][j]=(dp[i][j]+dp[i][k]-dp[k+1][j])%Mod;
else
dp[i][j]=(dp[i][j]+dp[i][k]*dp[k+1][j]%Mod)%Mod;
}
dp[i][j]=(dp[i][j]%Mod+Mod)%Mod;
dp[i][j]=dp[i][j]*fp(len,Mod-2)%Mod;
}
}
printf("%I64d\n",dp[1][n]*f[n-1]%Mod);
}
return 0;
}