楼兰图腾树状数组求逆序数

手可摘星辰！！

于 2019-08-12 14:24:04 发布

阅读量121

点赞数

分类专栏：数据结构——树状数组

本文链接：https://blog.csdn.net/flymoyu/article/details/99300549

版权

数据结构——树状数组专栏收录该内容

6 篇文章 0 订阅

订阅专栏

这个博客很详细
https://blog.csdn.net/ssimple_y/article/details/53744096

#include <cstdio>
#include <cstring>
#include <iostream>
#define N 200001
using namespace std;

int n,lh[N],ll[N],rh[N],rl[N],a[N],c[N];
long long ansup,ansdown;

int ask(int x)
{
	int sum=0;
	for (;x;x-=x&(-x)) sum+=c[x];
	return sum;
}

void add(int x)
{
	for (;x<=n;x+=x&(-x)) c[x]++;
}

int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		ll[i]=ask(a[i]-1);

		lh[i]=i-1-ll[i];
		//cout<<ll[i]<<" "<<lh[i]<<endl;
		add(a[i]);
	}
	memset(c,0,sizeof(c));
	for (int i=n;i>=1;i--)
	{
		rl[i]=ask(a[i]-1);
		rh[i]=n-i-rl[i];
		add(a[i]);
	}
	for (int i=2;i<n;i++)
	{
		ansup+=(long long)lh[i]*rh[i];
		ansdown+=(long long)ll[i]*rl[i];
	}
	cout<<ansup <<" "<<ansdown<<endl;
	return 0;
}