洛谷 P6046 纯粹容器 senpai 题解

题目传送门

题目大意： 给出 $n$ 个物品，第 $i$ 个物品强度为 $a_i$，每次随机选择相邻两个进行决斗，强度大的活，小的就没了。问每个物品期望活多少轮。

题解

很容易想到的是，每个物品的生死，取决于什么时候遇到它左边第一个比他大的或右边第一个比他大的物品。

考虑第 $x$ 个物品，设左边第一个比他大的位置在 $pre$，右边第一个比他大的在 $next$。

设 $s[i]$ 表示物品 $x$ 至少活了 $i$ 轮的概率，他等于 $1-$ 物品x在i-1轮内去世的概率。

物品 $x$ 要在 $i$ 轮内去世，要么就是碰到 $pre$，要么就是碰到 $next$ 了，做一下容斥概率就是：
$$\frac{C_{n-1-(x-pre)}^{i-(x-pre)}+C_{n-1-(next-x)}^{i-(next-x)}-C_{n-1-(next-pre)}^{i-(next-pre)}}{C_{n-1}^i}$$

$C_{n-1-(x-pre)}^{i-(x-pre)}$ $C_i^{next-pre}$ 就是前面两种情况的重叠部分，也就是 $pre$ 到 $next$ 这一段全部去世了。

然后 $s[i]$ 累加起来就是 $x$ 总的存活轮数期望了，时间复杂度 $O(n^2)$。

我这题代码有点奇奇怪怪的，虽然能 $AC$……具体你们看代码：

#include <cstdio>
#define maxn 110
//不知道为什么，数组一定要开大一点，只开到60的话会WA后面几个点，也不知道哪里数组越界了……
#define mod 998244353

int n,a[maxn],ans[maxn];
int fac[maxn],inv_fac[maxn];
int ksm(int x,int y)
{
	int re=1;
	while(y)
	{
		if(y&1)re=1ll*re*x%mod;
		x=1ll*x*x%mod;y>>=1;
	}
	return re;
}
#define inv(x) ksm(x,mod-2)
void work()
{
	fac[0]=1;
	for(int i=1;i<=50;i++)fac[i]=1ll*fac[i-1]*i%mod;
	//虽然说数组开到了[0,109]，但是如果这里的50改成100会照样WA，很奇妙……
	inv_fac[50]=inv(fac[50]);
	for(int i=49;i>=0;i--)inv_fac[i]=1ll*inv_fac[i+1]*(i+1)%mod;
}
int C(int x,int y)
{
	if(x<y)return 0;
	return 1ll*fac[x]*inv_fac[y]%mod*inv_fac[x-y]%mod;
}
int pre,next;

int main()
{
	scanf("%d",&n); work();
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	for(int i=1;i<=n;i++)
	{
		pre=next=-1;
		for(int j=i-1;j>=1;j--)if(a[j]>a[i]){pre=j;break;}
		for(int j=i+1;j<=n;j++)if(a[j]>a[i]){next=j;break;}
		if(pre==-1&&next==-1)ans[i]=n-1;
		else if(pre==-1&&next!=-1)for(int j=1;j<n-1;j++)
			ans[i]=(ans[i]+(1-1ll*C(n-1-(next-i),j-(next-i))*inv(C(n-1,j))%mod+mod)%mod)%mod;
		else if(pre!=-1&&next==-1)for(int j=1;j<n-1;j++)
			ans[i]=(ans[i]+(1-1ll*C(n-1-(i-pre),j-(i-pre))*inv(C(n-1,j))%mod+mod)%mod)%mod;
		else for(int j=1;j<n-2;j++)
			ans[i]=(ans[i]+(1ll-1ll*((C(n-1-(i-pre),j-(i-pre))+C(n-1-(next-i),j-(next-i)))%mod-C(n-1-(next-pre),j-(next-pre))+mod)%mod*inv(C(n-1,j))%mod+mod)%mod)%mod;
		printf("%d ",ans[i]);
	}
}