CodeForces 61 E.Enemy is weak (树状数组)

题意：

给长度为n的数组
求i<j<k但a[i]>a[j]>a[k]的三元组数量

数据范围：
n<=1e6
a[i]<=1e9

思路：

这种题写过好几次了，但是又楞了，还是记录一下
显然枚举中间然后求两边!

记录左边比比当前数大的数的个数和右边比当前数小的数的个数

然后枚举中间的数，答案累加左边乘右边就行了

求左边比他小啥的用线段树或者树状数组就行了
由于树状数组好写我就用树状数组了

最后要注意的是开longlong

code:

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=1e6+5;
int a[maxm];
int xx[maxm];
int l[maxm],r[maxm];
int c[maxm];
int lowbit(int i){
    return i&-i;
}
void add(int i){
    while(i<maxm){
        c[i]++;
        i+=lowbit(i);
    }
}
int ask(int i){
    int ans=0;
    while(i){
        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;
}
signed main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        xx[i]=a[i];
    }
    //离散化
    sort(xx+1,xx+1+n);
    int num=unique(xx+1,xx+1+n)-xx-1;
    for(int i=1;i<=n;i++){
        a[i]=lower_bound(xx+1,xx+1+num,a[i])-xx;
    }
    //
    for(int i=1;i<=n;i++){
        l[i]=(i-1)-ask(a[i]);//左边比他大的数的个数
        add(a[i]);
    }
    memset(c,0,sizeof c);
    for(int i=n;i>=1;i--){
        r[i]=ask(a[i]-1);//右边比他小的数的个数
        add(a[i]);
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        ans+=l[i]*r[i];
    }
    cout<<ans<<endl;
    return 0;
}