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

题目链接
注意：每次求逆序或正序和都需要初始化

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <iostream>
#include <cmath>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
using namespace std;
typedef long long ll;
const int MAX=0x3f3f3f3f;
const int N=200010;
int n;
int a[N],c[N],l[N],r[N];
int ask(int x){
	int ans=0;
	while(x){
		ans+=c[x];
		x-=x&-x;
	}
	return ans;
} 
void add(int x,int y){
	while(x<=n){
		c[x]+=y;
		x+=x&-x;
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	memset(c,0,sizeof(c));   //每次求逆序或正序和都需要初始化 
	for(int i=n;i>=1;i--){
		r[i]=ask(a[i]-1);
		add(a[i],1);
	}
	memset(c,0,sizeof(c));
	for(int i=1;i<=n;i++){
		l[i]=ask(a[i]-1);
		add(a[i],1);
	}
	ll ans=0;
	for(int i=1;i<=n;i++){
		ans+=(ll)(i-1-l[i])*(n-i-r[i]); //i左边有i-1个数 
	}
	cout<<ans<<endl;
	ans=0;
	for(int i=1;i<=n;i++){
		ans+=(ll)l[i]*r[i];
	} 
	cout<<ans<<endl;
}