第20届上海大学程序设计联赛春季赛 F - 到底是多少分啊
链接:https://ac.nowcoder.com/acm/contest/33785/F
来源:牛客网
在旅途的终点,你终于看到了大魔王小 X。
小 X 准备玩一个游戏,这个游戏规则如下:
对于一个含有 n 个数的数组 a。小 X 必须执行如下操作 t 次:
选择数组中的某一个数,给它加上 1。
经过这样 t 次操作后得到的数组 b。小 X 的得分是数组 b 中所有数的乘积。
显然,在不同的操作过程后,最终可能获得不同的数组 b。我们称两个数组是不同的,当且仅当这两个数组中至少有一个数不同,即两个数组 p, q 不同当且仅当存在 i 使得 pi≠qi 。
如果小 X 在所有不同的数组 b 中随机取一个作为最终结果,那么他的期望得分是多少?
- 题目要求的是期望得分,等于对\(n\)个数进行\(t\)次操作,能带来的得分总和除以操作的种类数。操作的种类我们可以看做球盒模型(\(t\)相同的球放进\(n\)个不同的盒子,可空)答案为\(C_{n + t - 1}^{n - 1}\),于是只需要求得得分总和即可。
- 考虑令\(f_{i,j}\)表示对前\(i\)个数进行\(j\)次操作能获得的不同的得分之和,状态转移的时候我们枚举在第\(i\)个数进行次\(k\)次操作,即
\[f_{i,j} = \sum_{k = 0}^{j - 1}(f_{i-1,j - k}\times (a_i + k))
\]
很明显这个转移时朴素的\(n^3\),对于这个枚举当前操作几次的\(dp\),我们考虑把它的形式拆开并且这样写观察一下
\[f_{i,j} = f_{i-1,j}\times a_i +f_{i-1,j - 1}\times (a_i + 1) + f_{i-1,j - 2}\times (a_i + 2) + \cdots +f_{i-1,0}\times (a_i + j) \\ f_{i,j - 1} = f_{i-1,j - 1}\times a_i + f_{i-1,j-2}\times (a_i + 1)\cdots +f_{i-1,0}\times (a_i + j - 1)
\]
可得
\[f_{i,j} = f_{i-1,j}\times a_i + f_{i,j - 1} + \sum_{k = 0}^{j - 1}f_{i-1,k}
\]
这样维护一个前缀和就可以\(O(n^2)\)转移了。
注意一下初始化的问题,仅让\(f_{0,0} = 1\)即可,他来更新最开始的答案,其它的直接转移即可。
#include <bits/stdc++.h>
using namespace std;
const long long mod = 998244353;
long long f[3010][3010],a[3010],s[3010][3010],fac[10010];
long long fiv[10010],inv[10010];
void init() {
fac[0] = fac[1] = fiv[0] = fiv[1] = inv[1] = 1;
for(int i = 2;i <= 10000;i ++) {
fac[i] = fac[i - 1] * i % mod;
inv[i] = (mod - mod / i) * inv[mod % i] % mod;
fiv[i] = inv[i] * fiv[i - 1] % mod;
}
}
long long ksm(long long a,long long b) {
long long res = 1;
while(b) {
if(b & 1) res = res * a % mod;
a = a * a % mod;
b >>= 1;
}
return res;
}
long long invv(int x) {
return ksm(1ll * x,mod - 2);
}
long long C(int n,int m) {
return fac[n] * fiv[m] % mod * fiv[n - m] % mod;
}
int main() {
init();
// fac[0] = 1;
// for(int i = 1;i <= 10000;i ++) fac[i] = fac[i - 1] * i % mod;
int n,t; cin >> n >> t;
for(int i = 1;i <= n;i ++) {
cin >> a[i];
}
s[0][0] = f[0][0] = 1;
for(int i = 1;i <= n;i ++) {
f[i][0] = f[i - 1][0] * a[i] % mod;
s[i][0] = f[i][0];
}
for(int i = 1;i <= t;i ++) {
s[0][i] = (s[0][i - 1] + f[0][i]) % mod;
}
for(int i = 1;i <= n;i ++) {
for(int j = 1;j <= t;j ++) {
f[i][j] = f[i - 1][j] * a[i] % mod;
f[i][j] = (f[i][j] + f[i][j - 1]) % mod;
f[i][j] = (f[i][j] + s[i - 1][j - 1]) % mod;
}
for(int j = 1;j <= t;j ++) {
s[i][j] = (s[i][j - 1] + f[i][j]) % mod;
}
}
long long ans = f[n][t] * invv(C(n + t - 1,n - 1)) % mod;
cout << ans << '\n';
return 0;
}