重点不在于推式子的计数提

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(aii+1)
咕咕咕

如果对所有情况下的某一特性计数,就可以利用期望的线性性拆到每个最小的单位上,然后乘上填数方案数。

可以认为线段树就是一种类似分块的数据结构,也就是线段树上的一个节点 ( n o t e , l , r ) (note,l,r) (note,l,r) 就是一个块长为 r − l + 1 r-l+1 rl+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;
}
  • 9
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值