hdu 2492 Ping pong (樹狀數組)

理解對了了題意的話應該算是道簡單的樹狀數組。

題目要求裁判不能同時低於兩個人，或者同時高於兩人。就是求一個長度為3的順序序列。

這樣的話對於第i個數ai,我們可以求出左邊比ai大的數lmaxi,右邊比ai小的數rmini.相乘再加上反過來左邊小右邊大的情況就是結果。

#include <iostream>#include<cstdio>#include<cstring>#define maxm 100001#define clr(A,k) memset(A,(k),sizeof(A))#define LL long longusing namespace std;int C[maxm],A[maxm];int lmin[maxm],rmin[maxm],lmax[maxm],rmax[maxm];void init(){    for(int i=0;i<maxm;i++){        C[i]=0;        lmin[i]=0;        lmax[i]=0;        rmin[i]=0;        rmax[i]=0;    }}int lowbit(int x){    return x&(-x);}void update(int pos,int val){    while(pos<=maxm){        C[pos]+=val;        pos+=lowbit(pos);    }}int query(int x){    int ret=0;    while(x>0){        ret+=C[x];        x-=lowbit(x);    }    return ret;}int main(){    int T,N;    scanf("%d",&T);    while(T--){        scanf("%d",&N);        for(int i=1;i<=N;i++){            scanf("%d",&A[i]);        }        init();        for(int i=1;i<=N;i++){            update(A[i],1);            int tmp=query(A[i]-1);            lmin[i]=tmp;            lmax[i]=i-tmp-1;//返回左邊比A[i]大的數        }        clr(C,0);        for(int i=N,j=1;i>=1;i--,j++){            update(A[i],1);            int tmp=query(A[i]-1);            rmin[i]=tmp;            rmax[i]=j-tmp-1;        }        LL sum=0;        for(int i=1;i<=N;i++){            sum+=lmin[i]*rmax[i]+lmax[i]*rmin[i];        }        printf("%I64d\n",sum);    }    return 0;}