HDU 2492 Ping pong

http://acm.hdu.edu.cn/showproblem.php?pid=2492

#include<iostream> using namespace std; #define N 100001 #define M 20001 __int64 c[N]; int a[M]; __int64 L[M],R[M]; int lowbit(int x){ return x&(-x); } void update(int x,int k){ while(x<N){ c[x]+=k; x+=lowbit(x); } } __int64 getsum(int x){ __int64 r=0; while(x>0){ r+=c[x]; x-=lowbit(x); } return r; } int main(void){ int t,n; scanf("%d",&t); while(t--){ scanf("%d",&n); memset(c,0,sizeof(c)); for(int i=1;i<=n;i++) scanf("%d",&a[i]); for(int i=1;i<=n;i++){ L[i]=getsum(a[i]-1); update(a[i],1); } memset(c,0,sizeof(c)); for(int i=n;i>=1;i--){ R[i]=getsum(a[i]-1); update(a[i],1); } __int64 ans=0; for(int i=1;i<=n;i++){ ans+=L[i]*(n-i-R[i]); ans+=R[i]*(i-1-L[i]); } printf("%I64d/n",ans); } }  

照题目意思，一个人做裁判的场次数就等于他左边小于它数的个数与右边大于它的数的个数乘积，再加上左边大于而右边小于它的数个数乘积

我是用树状数组插入做了两次，从左边开始插入，L数组记录左边小于它的个数；从右边开始插入，R数组记录右边小于它的个数，最后乘乘加加就好