hdu 2492 Ping pong

题目大意：找三个元素ai,aj,ak,要求i<j<k,并且ai<aj<ak,或者ai>aj>ak.问这样的三元组一共有多少个。

暴力搜索是不行的，只能搜其中一个就是aj，求用比aj小的乘以比aj大的加上比aj大的乘以比aj小的。就是搜索的aj这个元素有的三元组个数。用树状数组。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <math.h>
using namespace std;
int  a[100009];
int  d[20009],min1[20009],max1[20009];
const int N=100000;
int lowbit(int x)
{
    return x&(-x);
}
void add(int x)
{
    while(x<=N)
    {
        a[x]++;
        x+=lowbit(x);
    }
}
int sum(int x)
{
    int ans=0;
    while(x>0)
    {
        ans+=a[x];
        x-=lowbit(x);
    }
    return ans;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
    int n;
    scanf("%d",&n);
    long long ans=0;
    memset(a,0,sizeof(a));
    memset(min1,0,sizeof(min1));
    memset(max1,0,sizeof(max1));
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&d[i]);
        add(d[i]);
        min1[i]=sum(d[i]-1);
    }
     memset(a,0,sizeof(a));
      for(int i=n;i>=1;i--)
    {
        max1[i]=sum(d[i]-1);
        add(d[i]);
    }
    for(int i=1;i<=n;i++)
    {
        ans+=min1[i]*(n-i-max1[i])+max1[i]*(i-1-min1[i]);
    }
    cout<<ans<<endl;
}
	return 0;
}