洛谷-纯粹容器-(概率期望+组合数+容斥)

P6046

题意：
就是给你n个容器，每个容器一个强度，小A会进行n-1次操作，每次随机选择两个相邻未被击倒的容器进行决斗，强度低的会被击倒并移除队伍。现在问你每个容器存活的轮数的期望。

思考：

1. 刚开始我甚至以为是区间dp的期望题，因为选择两个碰，就代表中间的都已经处理完了。但是想了想好像没法写。那么就单看一个容器，对他而言期望就是概率×轮数。但是这个第i轮就结束的概率实在不好求，这种就是思想简单，实现复杂的。
2. 所以这时候就要拿出来期望的公式了，对于第i个容器，E[x_i] = sum(P(x_i>=j))。所以暴力枚举存活的轮数，用1减去j轮前死亡的概率，全部累加起来就可以了。不过死亡的概率怎么算呢？这里就是找到每个容器左边第一个大于他的，右边第一个大于他的。这样求一下x到l[x]全部选的方案数/总方案数，右边同理。但是这样会多减去，那么把l[x]到r[x]的再加上来就可以了。
3. 这种题其实还是比较难的，不过要有这种思想，慢慢的推。对于为什么看右边第一个大于他的，他完全可以被右边别的大于他的击败啊。但是不管被哪个击败，最终哪个容器都会跑到右边第一个大于他的位置上，所以那些是无关因子。不用考虑。这种还是要多见题，多多理解把。

代码：

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;
const int mod = 998244353,inf = 1e18;
const int N = 2e5+10,M = 2010;

int T,n,m,k;
int va[N];
int l[N],r[N];
int fact[N],infact[N];

int ksm(int a,int b)
{
	int sum = 1;
	while(b)
	{
		if(b&1) sum = sum*a%mod;
		a = a*a%mod;
		b >>= 1;
	}
	return sum;
}

void init(int x)
{
	fact[0] = infact[0] = 1;
	for(int i=1;i<=x;i++) fact[i] = fact[i-1]*i%mod;
	infact[x] = ksm(fact[x],mod-2)%mod;
	for(int i=x-1;i>=1;i--) infact[i] = infact[i+1]*(i+1)%mod;
}

int C(int a,int b)
{
	if(a<b||b<0) return 0;
	return fact[a]*infact[b]%mod*infact[a-b]%mod;
}

int solve(int x)
{
	int ans = 0;
	for(int i=1;i<=n-1;i++)
	{
		int pl = 0,pr = 0,ps = 0;
		if(l[x]) pl = C(n-1-(x-l[x]),i-(x-l[x]))*ksm(C(n-1,i),mod-2)%mod;
		if(r[x]) pr = C(n-1-(r[x]-x),i-(r[x]-x))*ksm(C(n-1,i),mod-2)%mod;
		if(l[x]&&r[x]) ps =  C(n-1-(r[x]-l[x]),i-(r[x]-l[x]))*ksm(C(n-1,i),mod-2)%mod;
		ans = (ans+1-pl-pr+ps)%mod;
	}
	return (ans%mod+mod)%mod;
}

signed main()
{
	IOS;
	init(1e5+10);
	cin>>n;
	for(int i=1;i<=n;i++) cin>>va[i];
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<i;j++) if(va[j]>va[i]) l[i] = j;
		for(int j=n;j>i;j--) if(va[j]>va[i]) r[i] = j;
	}
	for(int i=1;i<=n;i++) cout<<solve(i)<<" ";
	return 0;
}

总结：
多多积累经验，多多思考。