2021牛客暑期多校训练营1 I题

I题: Increasing Subsequence

题目链接:https://ac.nowcoder.com/acm/contest/11166/I

题目大意

Alice和Bob在一个大小为 $n(1\le n\le 5000)$的排列 $P$上玩游戏，双方轮流选择数字。
每一轮中，当前玩家需要选择一个比之前双方选择过的元素都大的元素。并且若该玩家之前选择了元素 $P_i$，当前回合选择了 $P_j$，则必须满足 $j＞i$。若有多种选择，则将等概率地随机选择一个合法的元素。
第一回合的玩家会随机选择一个元素，第二回合的玩家会随机选择一个比第一回合选择的数字大的元素，没有合法操作时游戏结束。
球游戏结束时，双方的总游戏回合数期望，答案对998244353取模。

题解

由题意不难看出，选择过程中，每一步所选的元素应呈上升趋势，即每一步所选元素都比上一步大，且为当前最大。
考虑动态规划，对于期望，$dp$数组采取倒推，保证每次能简单算出当前操作的发生概率。定义 $d{p}_{i,j}$表示前两个回合Alice和Bob分别选择 $P_i,P_j$时(不记录上一回合是谁操作，因为上家可以通过上次所选元素大小比较判断)，剩余回合数期望。
对于 $d{p}_{i,j}$，它的状态应是其下一步操作可达到的状态期望总和除以可能数，再加一(当前到下一步需一回合)
若前两个回合Alice和Bob分别选择了$P_x,P_y$，枚举下一回合可选的$P_k$动态转移方程可表示为:

若 $P_x>P_y$(即Alice上一回合选择 $P_x$,则该回合为Bob操作更新y,设$c$表示下一步$y$可能的更新值数量，即Bob选择的可能数):
$$d{p}_{x,y}=1+\frac{1}{c}\sum _{k>y,P_k>P_x}d{p}_{x,k}$$
若 $P_x<P_y$(即Bob上一回合选择 $P_y$,则该回合为Alice操作更新x,设$c$表示下一步$x$可能的更新值数量，即Alice选择的可能数):
$$d{p}_{x,y}=1+\frac{1}{c}\sum _{k>x,P_k>P_y}d{p}_{k,y}$$
最终结果:$$ans=\frac{1}{n}\sum _{i=1}^{n}d{p}_{i,0}$$(即先手Alice选择每一个的元素的状态期望总和除以元素总个数)
期望不一定为整数，会需要逆元，可以提前计算打表$1$~$n$每个数对$mod$的逆元(除数只可能为$c$或$n$，而$1\le c＜n$)，需要进行除法运算时换为乘法

参考代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const long long MAXN=5e3+5,mod=998244353;
ll N,P[MAXN],dp[MAXN][MAXN],inv[MAXN],cnt[MAXN],sum[MAXN],ans;
ll powmod(ll x){
	ll ret=1,s=mod-2;//根据费马小定理计算逆元，使用快速幂
	while(s){
		if(s&1){
			ret=ret*x%mod;
		}
		x=x*x%mod;
		s>>=1;
	}
	return ret;
}
int main()
{
	cin>>N;
	inv[0]=1;
	for(int i=1;i<=N;i++)inv[i]=powmod(i),scanf("%lld",&P[i]);//在输入的循环内同时进行逆元打表
	for(int i=N;i>=1;i--){
		int s=0,c=0;
		for(int j=N;j>=0;j--){
			if(i==j)continue;
			if(P[i]>P[j]){
				dp[i][j]=(s*inv[c]+1)%mod;//更新，注意inv数组直接用逆元结果计算替换除法
				sum[j]=(sum[j]+dp[i][j])%mod;//累加更新x时的总和
				cnt[j]++;//更新x时的可能数加一
			}
			else{
				dp[i][j]=(sum[j]*inv[cnt[j]]+1)%mod;//更新，注意inv数组直接用逆元结果计算替换除法
				s=(s+dp[i][j])%mod;//累加更新y时的总和
				c++;//更新y时的可能数加一
			}
		}
	}
	for(int i=1;i<=N;i++)ans=(ans+dp[i][0])%mod;
	ans=(ans*inv[N])%mod;
	cout<<ans<<endl;
	return 0;
}