[UNR #3]百鸽笼

我好菜啊,就在网上看了一堆博(dai)客(ma),终于搞懂了一点

思路

因为每列满不满不好处理,用容斥。

\(L\)有空位等价于其他列都比\(L\)先满(限制)。

现在考虑一个列\(L\)的概率,有一个列集合\(S\)\(S\)中都要比\(i\)先空,\(S\)就相当于必须违反限制的集合。

\(S\)带上一个容斥系数\((-1)^{|S|}\),加起来就是答案。

可以DP计数\(S\)的方案数,发现容斥系数只和\(|S|\)有关,所以就不枚举\(S\)了,改成记录\(|S|\)和共进入了多少人\(len\)即可。

因为这个背包DP是计数,所以很容易撤销,后面具体讲。

DP的方法

这块是我花时间最多的地方,基本上其他地方都没讲。(可能是我太菜了吧)

\(f_{i,j}\)表示\(|S|=i\),总共有\(j\)个人(\(len=j\))的方案数。

最重要的就是要保证除了计算的那列可以放满,其他的都放不满(这样保证放的列数不变),所以\(cnt\)\(-1\),留出一个不能放!不保证这个,刚才的容斥就没有意义了。

\[f_{i +1, j + c} = f_{i, j} \cdot \binom{j+c}{c} \]

意思是这一列在\(L\)之前填了\(c\)\(c=0\)是允许的),然后把这\(c\)个插入操作序列的方案数。

最后再把\(L\)列的元素插入到排列中,注意只插\(a_L-1\)个,最后一个保证插到最后,因为\(L\)列满了后面的都不考虑,这样避免重复计数

最后,因为之前已经保证了不会在\(L\)列前把其他列选完(减了\(1\)个),所以每次都可以任选一列放入,因此总方案数就是\((|S|+ 1) ^ {len}\),就是分母了。
// 这个解释不是很合理,先咕了

怎么撤销

因为加一个物品是从后往前,每次用旧的值去更新成新的值。

所以撤销时要从前往后,从\(i=0, j=0\)开始(始终是原值),把新值再变回旧值,再往后更新。

Code

没有对任何边界情况做处理,所以常数非常大。

#include <cstdio>
#include <algorithm>
using namespace std;
#define File(s) freopen(s".in", "r", stdin), freopen(s".out", "w", stdout)
typedef long long ll;

const int N = 35, Sum = 935, p = 998244353;
inline int add(int x, int y){return x+y>=p ? x+y-p : x+y;}
inline int sub(int x, int y){return x-y<0 ? x-y+p : x-y;}
inline int mul(int x, int y){return 1LL * x * y % p;};
inline int power(int x, int y){
	int res = 1;
	for(; y; y>>=1, x = mul(x, x)) if(y & 1) res = mul(res, x);
	return res;
}
int a[N];
int n, sum = 0;
int powinv[N][Sum], C[Sum][Sum];
int f[N][Sum];

void pre(int n, int sum){
	for(int i=1; i<=n; i++){
		powinv[i][0] = 1; powinv[i][1] = power(i, p-2);
		// printf("powinv[%d][%d] = %d\n", i, 1, powinv[i][1]);
		for(int j=2; j<=sum; j++)
			powinv[i][j] = mul(powinv[i][j-1], powinv[i][1]);
	}
	C[0][0] = 1;
	for(int i=1; i<=sum; i++){
		C[i][0] = 1;
		for(int j=1; j<=i; j++) C[i][j] = add(C[i-1][j], C[i-1][j-1]);
	}
}

void pack(int cnt){
	for(int i=n-1; i>=0; i--)
		for(int j=sum; j>=0; j--)
			for(int c=0; c<cnt; c++) // 不能全进
				f[i+1][j+c] = sub(f[i+1][j+c], mul(f[i][j], C[j+c][c]));
}
void unpack(int cnt){
	for(int i=0; i<n; i++)
		for(int j=0; j<=sum; j++)
			for(int c=0; c<cnt; c++)
				f[i+1][j+c] = add(f[i+1][j+c], mul(f[i][j], C[j+c][c]));
}

int main(){
	scanf("%d", &n);
	for(int i=1; i<=n; i++) scanf("%d", a + i), sum += a[i];
	pre(n, sum);
	f[0][0] = 1;
	for(int i=1; i<=n; i++) pack(a[i]);
	for(int i=1; i<=n; i++){
		unpack(a[i]);
		int res = 0;
		for(int j=0; j<n; j++)
			for(int k=0; k<=sum - a[i]; k++)
				res = add(res, mul(powinv[j+1][k + a[i]], mul(f[j][k], C[k + a[i] - 1][a[i] - 1])));//, printf("res = %d\n", res);
		printf("%d ", res);
		pack(a[i]);
	}
	return 0;
}
posted @ 2020-01-28 23:40  RiverHamster  阅读(233)  评论(0编辑  收藏  举报
\