CF1096F Inversion Expectation [期望]

题意

  给你一个长为$n$的排列,若某一位为$−1$则这一位是不确定的。每种可能的排列出现的概率相等。求期望逆序对数对$998244353$取模的结果。


分类讨论一下:

  1. 已知位置之间贡献:树状数组计算

  2.未知位置之间贡献:$\frac{\tbinom{m}{2}}{2}=\frac{1}{4}m(m-1)$ ($m$为未知位置个数)

  3.已知和未知之间贡献:

    对于每个未知的位置,取不同值时前面比它大和后面比它小的数的个数。

    转化一下就是:每个已知的位置贡献为$$(前面未知位置个数\times比这个数大的未使用的数的个数+后面未知位置个数\times比这个数小的未使用的数的个数)\times m^{-1}$$

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 using namespace std;
 5 typedef long long ll;
 6 const int N=200010;
 7 const ll P=998244353;
 8 ll n,a[N],app[N],b[N],m,c[N],inv,S;
 9 ll ans;
10 inline void add(int x) {for (; x<=n; x+=(x&-x)) c[x]++;}
11 inline ll sum(int x) {ll s=0; for (; x; x-=(x&-x)) s+=c[x]; return s;}
12 inline ll pow(ll x,ll k) {ll r=1; for (; k; k>>=1,x=x*x%P) if (k&1) r=r*x%P; return r;}
13 int main() {
14     scanf("%lld",&n);
15     for (int i=1; i<=n; i++) {scanf("%lld",&a[i]),m+=(a[i]==-1); if (~a[i]) app[a[i]]=1;}
16     for (int i=1; i<=n; i++) {
17         b[i]=b[i-1]+(app[i]^1);
18         if (~a[i]) {
19             ans+=sum(n)-sum(a[i]);
20             add(a[i]);
21         }
22     }
23     ans=(ans%P)+1ll*m*(m-1)%P*748683265ll%P; //4^-1
24     inv=pow(m,P-2); S=0;
25     for (int i=1; i<=n; i++) {
26         if (~a[i]) S+=b[a[i]];
27         else ans=(ans+1ll*S*inv%P)%P;
28     }
29     S=0;
30     for (int i=n; i; i--) {
31         if (~a[i]) S+=m-b[a[i]];
32         else ans=(ans+1ll*S*inv%P)%P;
33     }
34     printf("%lld\n",ans);
35     return 0;
36 }

 

猜你喜欢

转载自www.cnblogs.com/hnooo/p/10207651.html