AGC023E Inversions
题目
首先,整个
{
a
}
\{a\}
{a} 的填数方案是经典的
a
a
a 数组从小到大排好序后
∏
i
=
1
n
(
a
i
−
i
+
1
)
\prod_{i=1}^n (a_i-i+1)
∏i=1n(ai−i+1)。
咕咕咕
如果对所有情况下的某一特性计数,就可以利用期望的线性性拆到每个最小的单位上,然后乘上填数方案数。
可以认为线段树就是一种类似分块的数据结构,也就是线段树上的一个节点 ( n o t e , l , r ) (note,l,r) (note,l,r) 就是一个块长为 r − l + 1 r-l+1 r−l+1 的维护着区间 [ l , r ] [l,r] [l,r] 的块。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<map>
#include<queue>
#define lc note<<1
#define rc note<<1|1
#define mp make_pair
using namespace std;
const int N=2e5+5;
const int mod=1e9+7;
int n,tot=1,ans;
int a[N],id[N],val[N];
pair<int,int> c[N];
inline int read()
{
int s=0,t=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')t=-1;ch=getchar();}
while(ch>='0'&&ch<='9') s=(s<<1)+(s<<3)+(ch^48),ch=getchar();
return s*t;
}
inline int inv(int i)
{
if(i<=1) return 1;
return 1ll*(mod-mod/i)*inv(mod%i)%mod;
}
struct sgt
{
struct Tree
{
int sum,cnt,tag;
}s[N<<2];
inline void build(int note,int l,int r)
{
s[note].sum=s[note].cnt=0;
s[note].tag=1;
if(l==r) return;
int mid=(l+r)>>1;
build(lc,l,mid);
build(rc,mid+1,r);
return;
}
inline void push_up(int note)
{
s[note].cnt=s[lc].cnt+s[rc].cnt;
s[note].sum=(s[lc].sum+s[rc].sum)%mod;
return;
}
inline void push_down(int note)
{
if(s[note].tag==1) return;
s[lc].sum=1ll*s[lc].sum*s[note].tag%mod;
s[rc].sum=1ll*s[rc].sum*s[note].tag%mod;
s[lc].tag=1ll*s[lc].tag*s[note].tag%mod;
s[rc].tag=1ll*s[rc].tag*s[note].tag%mod;
s[note].tag=1;
return;
}
inline void mul(int val)
{
s[1].sum=1ll*s[1].sum*val%mod;
s[1].tag=1ll*s[1].tag*val%mod;
return;
}
inline void modify(int note,int l,int r,int pos,int val)
{
if(l==r)
{
s[note].sum=val;
s[note].cnt=1;
return;
}
push_down(note);
int mid=(l+r)>>1;
if(pos<=mid) modify(lc,l,mid,pos,val);
else modify(rc,mid+1,r,pos,val);
push_up(note);
return;
}
inline int querycnt(int note,int l,int r,int ql,int qr)
{
if(ql>qr) return 0;
if(ql<=l&&r<=qr) return s[note].cnt;
push_down(note);
int mid=(l+r)>>1,res=0;
if(ql<=mid) res+=querycnt(lc,l,mid,ql,qr);
if(qr>mid) res+=querycnt(rc,mid+1,r,ql,qr);
return res;
}
inline int querysum(int note,int l,int r,int ql,int qr)
{
if(ql>qr) return 0;
if(ql<=l&&r<=qr) return s[note].sum;
push_down(note);
int mid=(l+r)>>1,res=0;
if(ql<=mid) res=(res+querysum(lc,l,mid,ql,qr))%mod;
if(qr>mid) res=(res+querysum(rc,mid+1,r,ql,qr))%mod;
return res;
}
}T;
int main()
{
n=read();
for(int i=1;i<=n;++i)
{
a[i]=read();
c[i]=mp(a[i],i);
}
sort(c+1,c+n+1);
for(int i=1;i<=n;++i)
{
int val=c[i].first;
int cur=c[i].second;
tot=1ll*tot*(val-i+1)%mod;
id[cur]=i;
}
for(int i=1;i<=n;++i) val[i]=1ll*(a[i]-id[i])*inv(a[i]-id[i]+1)%mod;
T.build(1,1,n);
for(int i=n;i;--i)
{
int cur=c[i].second;
ans=(ans+1ll*tot*(a[cur]-i)%mod*inv(2)%mod*T.querysum(1,1,n,cur+1,n)%mod)%mod;
int cnt=T.querycnt(1,1,n,1,cur-1);
int qwq=1ll*tot*(a[cur]-i)%mod*inv(2)%mod*T.querysum(1,1,n,1,cur-1)%mod;
ans=(ans+1ll*cnt*tot%mod-qwq+mod)%mod;
T.mul(val[cur]);
T.modify(1,1,n,cur,inv(a[cur]-i+1));
}
printf("%d",ans);
return 0;
}