Codeforces Round #829 C. Wish I Knew How to Sort

 题意：给你一个长为n的0/1序列，一次操作随机大概率的选两个下标i和j（i<j），如果ai>aj，其实就是ai是1，aj是0，那么就交换ai和aj。问使得序列被排好序的期望操作次数。

这题难就难在想到一种状态表示，能把结果算出来。

像101010001，你最终的目标是变成000001111，我们注意到前5个都得是0，现在前5个里有3个1

我们当前状态距离最终的状态还有3个位置需要从1变成0，可以想到定义f【i】表示还有i个这样需要变的位置的状态到最终状态的期望步数。

最终状态的值很好算，就是f【0】=0

f【i】这些状态操作一次可能变成f【i-1】和f【i】，变成f【i-1】的概率p是2*i*i/n*（n-1）

那么f【i】=p*f【i-1】+（1-p）*f【i】+1

化简为f【i】=f【i-1】+1/p

实现看代码

#include<bits/stdc++.h>
#define int long long
#define rep(i,a,n) for(int i=a;i<n;i++)
#define per(i,a,n) for(int i=n-1;i>=a;i--)
#define fi first
#define se second
#define pb push_back
#define endl '\n'
using namespace std;
const int inf = 0x3f3f3f3f;
const int mod = 998244353;
void add(int &a,int b){
    a+=b;
    if(a>=mod) a-=mod;
}
int qmi(int a,int b){
    int res=1;
    for(;b;b>>=1){
        if(b&1) res=res*a%mod;
        a=a*a%mod;
    }return res;
}
int f[200010];
void solve(){
    int n;
    cin>>n;
    vector<int> v(n);
    rep(i,0,n) cin>>v[i];
    vector<int> temp(v);
    sort(v.begin(),v.end());
    int st=0;
    rep(i,0,n){
        if(v[i]==0&&temp[i]) st++;
    }
    f[0]=0;
    rep(i,1,st+1){
        f[i]=(f[i-1]+n*(n-1)%mod*qmi(2ll*i%mod*i%mod,mod-2)%mod)%mod;
    }
    cout<<f[st]<<endl;

}
signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    int t;
    cin>>t;
    while(t--){
        solve();
    }
}