原来上一题是多校第二道期望题啊,我怎么没印象
题目链接:https://ac.nowcoder.com/acm/contest/11166/I
这题转移就比上题容易了些,但这题的状态比较难找到。
提取题目中的关键点:p是一个排列,每次选择的数需要大于所有已选的数。因为是一个排列,所以我们可以把当前选的数的大小作为状态。
因此,我们可以尝试状态dp[i][j]:上一轮选择了i,前一轮选择了j时的期望轮数。
之后,据次得出状态转移式:
ps.这里式根据 (可能变成的状态的dp值和+单步贡献)*变成该状态的概率 值得到的。
具体看代码吧()
#include <bits/stdc++.h>
#define inf 0x3f3f3f3f
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);
#define rep(i, a, n) for(int i = a; i <= n; ++ i)
#define per(i, a, n) for(int i = n; i >= a; -- i)
//#define ONLINE_JUDGE
using namespace std;
typedef long long ll;
const int mod=998244353;
template<typename T>void write(T x)
{
if(x<0)
{
putchar('-');
x=-x;
}
if(x>9)
{
write(x/10);
}
putchar(x%10+'0');
}
template<typename T> void read(T &x)
{
x = 0;char ch = getchar();ll f = 1;
while(!isdigit(ch)){if(ch == '-')f*=-1;ch=getchar();}
while(isdigit(ch)){x = x*10+ch-48;ch=getchar();}x*=f;
}
int gcd(int a,int b){return b==0?a:gcd(b,a%b);}
int lcm(int a,int b){return a/gcd(a,b)*b;};
ll ksm(ll a,ll n){
ll ans=1;
while(n){
if(n&1) ans=(ans*a)%mod;
a=a*a%mod;
n>>=1;
}
return ans%mod;
}
//==============================================================
const int maxn=5e3+10;
#define int ll
int n,p[maxn],q[maxn];
int cnt[maxn],sum[maxn];
int Inv[maxn];
int dp[maxn][maxn];
int inv(int x){
return Inv[x];
}
signed main()
{
#ifndef ONLINE_JUDGE
freopen("in.txt","r",stdin);
freopen("out.txt","w",stdout);
#endif
//clock_t c1 = clock();
//===========================================================
read(n);
for(int i=0;i<maxn;++i)Inv[i]=ksm(i,mod-2);
for(int i=1;i<=n;++i)read(p[i]),q[p[i]]=i;
for(int i=n;i>=1;--i){
memset(cnt,0,sizeof(cnt));
memset(sum,0,sizeof(sum));
for(int j=i+1;j<=n;++j){
sum[q[j]]+=dp[i][j];
sum[q[j]]%=mod;
cnt[q[j]]+=1;
}
for(int j=n-1;j>=0;--j){
sum[j]+=sum[j+1];
sum[j]%=mod;
cnt[j]+=cnt[j+1];
}
/* for(int j=0;j<=n;++j){
cerr<<sum[j]<<" "<<cnt[j]<<endl;
}
cerr<<"================================"<<endl; */
for(int j=0;j<i;++j){
int num=cnt[q[j]];
int s=sum[q[j]];
if(num){
dp[j][i]=(dp[j][i]+s*inv(num)%mod+1)%mod;
}
}
}
int res=0;
for(int i=1;i<=n;++i){
res+=dp[0][i];
res%=mod;
}
res=(res*inv(n))%mod;
res=(res+1)%mod;
cout<<res<<endl;
//===========================================================
//std::cerr << "Time:" << clock() - c1 << "ms" << std::endl;
return 0;
}

京公网安备 11010502036488号