[ARC104D] Multiset Mean

考虑计算和为 \(x\) 的方案时,把所有的数减去 \(x\),dp 出和等于 \(0\) 的。
减去后数被分为三段,小于 \(0\),等于 \(0\) 和大于 \(0\)
其中等于 \(0\) 的直接乘上即可,对于正负,上下都是对称的,
直接 dp 出 \(f_{i,j}\) 表示用了前 \(i\) 个数和为 \(j\) 的方案书,
使用前缀和优化,最后正负乘起来相加即可。

const int N = 1e2 + 5;
int n, k, m;
int f[N][N * N * N], ans[N];
void solve() {
	cin >> n >> k >> m;
	ll s = 0;
	f[0][0] = 1;
	FOR(i, 1, n) {
		FOR(j, 0, s) f[i][j] = f[i - 1][j];
		s += i * k;
		FOR(j, i, s) Add(f[i][j], f[i][j - i]);
		int c = i * (k + 1);
		ROF(j, s, c) Sub(f[i][j], f[i][j - c]); 
	}
	FOR(i, 1, n) {
		FOR(j, 0, s) {
			Add(ans[i], mul(f[i - 1][j], f[n - i][j]));
		}
		Mul(ans[i], k + 1);
	}
	FOR(i, 1, n) cout << sub(ans[i], 1) << endl;
}
posted @ 2024-02-22 12:41  KevinLikesCoding  阅读(7)  评论(0编辑  收藏  举报