楼兰图腾题解（树状数组解法）

描述

在完成了分配任务之后，西部314来到了楼兰古城的西部。相传很久以前这片土地上(比楼兰古城还早)生活着两个部落，一个部落崇拜尖刀(‘V’)，一个部落崇拜铁锹(‘∧’)，他们分别用V和∧的形状来代表各自部落的图腾。
西部314在楼兰古城的下面发现了一幅巨大的壁画，壁画上被标记出了N个点，经测量发现这N个点的水平位置和竖直位置是两两不同的。西部314认为这幅壁画所包含的信息与这N个点的相对位置有关，因此不妨设坐标分别为(1,y1),(2,y2),…,(n,yn),其中y1~yn是1到n的一个排列。

西部314打算研究这幅壁画中包含着多少个图腾，其中V图腾的定义如下（注意：图腾的形式只和这三个纵坐标的相对大小排列顺序有关）1<=i<j<k<=n且yi>yj,yj<yk;

而崇拜∧的部落的图腾被定义为1<=i<j<k<=n且yi<yj,yj>yk;

西部314想知道，这n个点中两个部落图腾的数目。因此，你需要编写一个程序来求出V的个数和∧的个数。

输入格式

第一行一个数n
第二行是n个数，分别代表y1，y2……yn

输出格式

两个数
中间用空格隔开
依次为V的个数和∧的个数

样例输入

5
1 5 3 2 4

样例输出

3 4

数据范围与约定

10%的数据 n<=600
40%的数据 n<=5000
100%的数据 n<=200000，答案不超过int64

解题思路：

  本题的要求很显然，在给出的点中，可以构成多少个倒角A，可以构成多少V。显然，由于x坐标是升序给出的，所以我们如果知道某个点 i 左边有多少点的 y < yi ，右边有多少点 y < yi，那么可以构成的V就是这两个数之积，而最终答案ans就是所有位置积的累加。同理A符号也可以根据左右边各有多少点高于它来求。
  而我们可以根据树状数组来求逆序对，具体原理是：

• 查询前缀和[ a1, a[i-1] ]，累加到逆序对个数num上。
• 执行“单点增加”操作，即把位置 a[i] 上的数+1，同时正确维护前缀和（c[ i ] 存放的是 i 出现的次数，因此 i 的前缀和就是小于 i 的数的个数）。

代码示例：

#include<iostream>
#include<cstdio>
#include<cstring> 
using namespace std;
const int maxn = 200020;
typedef long long ll;
int a[maxn],c[maxn],lowbit[maxn]; //这里的c[x]记录x出现次数 
ll ldown[maxn],rdown[maxn];	//ldown[x]表示x左边有几个比他矮的，rdown同理 
ll lup[maxn],rup[maxn];		//同上 
int n;
void add(int x,int y){
	for(;x <= n;x += lowbit[x]) c[x] += y;
}
ll ask(int x){
	ll ans = 0;
	for(;x;x -= lowbit[x]) ans += c[x];
	return ans;
}
void solve(){
	for(int i = 1;i <= n;i++){
		ldown[i] += ask(a[i]-1);
		add(a[i],1);
	}
	memset(c,0,sizeof c);
	for(int i = n;i > 0;i--){
		rdown[i] += ask(a[i]-1);
		add(a[i],1); 
	}
	ll ans = 0;
	for(int i = 1;i <= n;i++) ans += ldown[i]*rdown[i];
	memset(c,0,sizeof c);
	for(int i = 1;i <= n;i++){
		lup[i] += ask(n) - ask(a[i]);
		add(a[i],1);
	}
	memset(c,0,sizeof c);
	for(int i = n;i > 0;i--){
		rup[i] += ask(n) - ask(a[i]);
		add(a[i],1); 
	}
	ll ans2 = 0;
	for(int i = 1;i <= n;i++) ans2 += lup[i]*rup[i];
	printf("%lld %lld\n",ans2,ans);
}
int main(){
	for(int i = 1;i < maxn;i++) lowbit[i] = i&-i;
	scanf("%d",&n);
	for(int i = 1;i <= n;i++)
		scanf("%d",a+i);
	solve(); 
	return 0;
}