左方之地
阿新 • • 發佈:2019-01-23
題目大意:
給定一個序列。
問一棵Treap(點的編號是有堆性質的二叉樹)的期望值是多少。
題解:
有個結論是,一個點的深度是這樣的:
將其編號的中序遍歷寫下來,從這個點對應中序遍歷的位置向左走,一開始計數器是0,每次遇到一個更小的數字就計數器++,向右同理,最後再加個1.
那麼考慮期望的線性性,算x對答案的貢獻,等價於計算x>y對答案有貢獻的排列有多少,然後+1。
不妨欽定x在y的左邊(最後還要乘以2),列舉中間隔著幾個位置d,那麼比y小的數字都要放在兩側,剩餘的隨意選:
最後那一項是列舉x的位置。
後半部分:
因此:
最後答案就是:
就做完了。
upd:其實後面還可以繼續稍微化簡一下:
不過沒啥本質區別就是了……
#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define Rep(i,v) rep(i,0,(int)v.size()-1)
#define lint long long
#define mod 998244353
#define db long double
#define pb push_back
#define mp make_pair
#define fir first
#define sec second
#define gc getchar()
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
typedef pair<int,int> pii;
typedef set<int>::iterator sit;
inline int inn()
{
int x,ch;while((ch=gc)<'0'||ch>'9');
x=ch^'0';while((ch=gc)>='0'&&ch<='9')
x=(x<<1)+(x<<3)+(ch^'0');return x;
}
const int N=100010;
int fac[N],facinv[N];
inline int C(int n,int m) { if(n<0||m<0||n<m) return 0;return (lint)fac[n]*facinv[n-m]%mod*facinv[m]%mod; }
inline int fast_pow(int x,int k,int ans=1) { for(;k;k>>=1,x=(lint)x*x%mod) (k&1)?ans=(lint)ans*x%mod:0;return ans; }
inline int prelude(int n)
{
rep(i,fac[0]=1,n) fac[i]=(lint)fac[i-1]*i%mod;
facinv[n]=fast_pow(fac[n],mod-2);
for(int i=n-1;i>=0;i--) facinv[i]=(i+1ll)*facinv[i+1]%mod;
return 0;
}
int a[N],f[N];
int main()
{
int n=inn(),ans=0;prelude(n);rep(i,1,n) a[i]=inn()%mod;f[1]=0;
rep(x,2,n) f[x]=(f[x-1]+(lint)fac[n-1]*fast_pow(C(n-1,x-1),mod-2)%mod*C(n,x)%mod)%mod;
rep(x,1,n) ans=(ans+a[x]*(2ll*f[x]%mod*facinv[n]%mod+1)%mod)%mod;
return !printf("%d\n",ans);
}