ch4201 楼兰图腾 树状数组求逆序对（双向）

描述

在完成了分配任务之后，西部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\

分析：

首先要回会用树状数组求逆序对：这里
正向扫描一次求出左边比当前点大的点的数量left[i]
（还有比当前点小的点的数量）
反向扫描一次求出右边比当前点大的点的数量right[i]
（还有比当前点小的点的数量）
然后枚举每个点
求v型 就是ans+=left[i]* right[i]
另一个类似
具体看代码

ps：
过了前几个案例之后后面都是wa的
最后发现乘法过程会爆int，我都惊了，不是说好答案不超过int吗
仔细一看是不超过int64。。也就是long long。有毒

code：

#include<iostream>
#include<cstdio>
typedef long long ll;
const int inf=0x3f3f3f3f;
const int inn=0x80808080;
using namespace std;
const int maxm=2e5+5;
int a[maxm];
int c[maxm];
int n;
int l[maxm];
int r[maxm];
int l2[maxm];
int r2[maxm];
int lowbit(int x){
    return x&(-x);
}
void add(int i,int t){
    while(i<maxm){
        c[i]+=t;
        i+=lowbit(i);
    }
}
int sum(int i){
    int res=0;
    while(i){
        res+=c[i];
        i-=lowbit(i);
    }
    return res;
}
int main(){
    while(cin>>n){
        memset(c,0,sizeof c);
        for(int i=1;i<=n;i++){
            cin>>a[i];
            add(a[i],1);
            l[i]=i-sum(a[i]);//求左边比他大的点 的数量
            l2[i]=sum(a[i]-1);//求左边比他小的 的数量
        }
        memset(c,0,sizeof c);
        for(int i=n;i>=1;i--){
            add(a[i],1);
            r[i]=(n-i+1)-sum(a[i]);//求右边比他大的点 的数量
            r2[i]=sum(a[i]-1);//求右边比他小的点 的数量
        }
        ll ans1=0,ans2=0;
        for(int i=1;i<=n;i++){
            ans1+=(long long)l[i]*r[i];//这里乘法爆int，贼坑
            ans2+=(long long)l2[i]*r2[i];
        }
        cout<<ans1<<' '<<ans2<<endl;
    }
    return 0;
}