91pts #3RE求助
查看原帖
91pts #3RE求助
177070
暗ざ之殇楼主2020/7/9 20:47

RT

Code:

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
const int N=4e6;
int n,m;
int sum[N<<2],L[N],R[N];
long long ans;
struct node
{
	int num,id,now;
}a[N];
bool cmp1(node x,node y)
{
	return x.num<=y.num;
}
bool cmp2(node x,node y)
{
	return x.id<y.id;
}
void update(int node)
{
	sum[node]=sum[node<<1]+sum[node<<1|1];
}
void insert(int node,int l,int r,int x,int k)
{
	if(l==r)
	{
		sum[node]+=k;
		return ;
	}
	int mid=(l+r)>>1;
	if(x<=mid) insert(node<<1,l,mid,x,k);
	else insert(node<<1|1,mid+1,r,x,k);
	update(node);
}
int query(int node,int l,int r,int x,int y)
{
	if(x<=l&&r<=y) return sum[node];
	int mid=(l+r)>>1;
	int cnt=0;
	if(x<=mid) cnt+=query(node<<1,l,mid,x,y);
	if(y>mid) cnt+=query(node<<1|1,mid+1,r,x,y);
	return cnt;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i].num);
		a[i].id=i;
	}
	sort(a+1,a+1+n,cmp1);
	for(int i=1;i<=n;i++)
	{
		if(a[i].num>a[i-1].num) m++;
		a[i].now=m;
	}
	sort(a+1,a+1+n,cmp2);
	for(int i=1;i<=n;i++)
	{
		insert(1,1,n,a[i].now,1);
		if(a[i].now==1) L[i]=0;
		else L[i]=query(1,1,n,1,a[i].now-1);
	}
	memset(sum,0,sizeof(sum));
	for(int i=n;i;i--)
	{
		insert(1,1,n,a[i].now,1);
		if(a[i].now==n) R[i]=0;
		else R[i]=query(1,1,n,a[i].now+1,n);
	}
	for(int i=1;i<=n;i++)
	    ans+=L[i]*R[i];
	printf("%lld\n",ans);
	return 0;
}
2020/7/9 20:47
加载中...