HDU 2492 pingpang(树状数组)

 问题描述:给出一列数Ai，任选三个上升或下降的数为一组，问可以有这样的多少组。

 数据范围：0<i<20000,ai<100000

 分析：我觉得。。打死我也想不出这个用树状数组。。

           问题转化成 求每个数左右两边比它小的数。因为扫一遍就已经n了。。再累计比它小的数，说明是遍扫描边更新。树状数组求和和改变的操作复杂度为O（log ai）。扫描到每个数时，我们可以计算sum(a[i]-1)，再add(a[i],1).逆序同样处理一遍就可以了。

 code：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
const int maxai=100000+10;
const int maxn=20000+10;


int leftles[maxn],rightles[maxn],a[maxn];
int leftless[maxai],rightless[maxai];

void init(){
memset(leftles,0,sizeof(leftles));
memset(a,0,sizeof(a));
memset(rightles,0,sizeof(rightles));
memset(leftless,0,sizeof(leftless));
memset(rightless,0,sizeof(rightless));
}
int lowbit(int x){ return x&-x; }
int sum(int x,int *s){
 int ret=0;
 while(x>0){
    ret+=s[x];
    x-=lowbit(x);
 }
 return ret;
}

void add(int x,int d,int *s){
  while(x<=maxai){
    s[x]+=d;
    x+=lowbit(x);
  }
}



int main(){
int T;
 scanf("%d",&T);
 while(T--){
    int n;
    scanf("%d",&n);
    init();
    for(int i=0;i<n;i++) scanf("%d",&a[i]);

    for(int i=0;i<n;i++){
        leftles[i]=sum(a[i]-1,leftless);
        add(a[i],1,leftless);
    }
    for(int i=n-1;i>=0;i--){
        rightles[i]=sum(a[i]-1,rightless);
        add(a[i],1,rightless);
    }
    long long ans=0;
    for(int i=0;i<n;i++){
      ans=ans+(i-leftles[i])*rightles[i]+leftles[i]*(n-1-i-rightles[i]);
    }
    printf("%lld\n",ans);
 }
 return 0;
}