LA 4329 ping-pong树状数组

题目链接：

刘汝佳，大白书，P197.

枚举裁判的位置，当裁判为i时，可以有多少种选法，如果已经知道在位置i之前有ci个数比ai小，那么在位置i之前就有i-1-ci个数比ai大。

在位置i之后有di个数比ai小，那么在位置i之后就有n-i-di个数比ai大。

这样，选法数有ci*(n-i-di)+di*(i-1-ci).

注意到ai的值各不相同且ai的值最大不超过10^5，那样就可以用v[a[i]]表示a[i]是否已经存在，这样当顺时针扫描的时候，每次更新v[a[i]] = 1;//表示已经存在。

比ai小的数的个数就是sum(v[j]) j<a[i].

至于d[i]的求法逆着扫描即可，如果用树状数组加速。

贴代码：

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 using namespace std;
 typedef long long int ll;
 const int N1 =,N2 = ;
 int a[N1],c[N2],sum[N1];
 int lowbit(int x)
 {
     return x&(-x);
 }
 void add(int index,int value)
 {
     while(index < N2)
     {
         c[index] += value;
         index += lowbit(index);
     }
 }
 int getSum(int index)
 {
     int s =;
     while(index)
     {
         s += c[index];
         index -= lowbit(index);
     }
     return s;
 }
 int main()
 {
 //    freopen("in.txt","r",stdin);
     int T,n;
     scanf("%d",&T);
     while(T--)
     {
         scanf("%d",&n);
         for(int i=; i<=n; ++i)
             scanf("%d",&a[i]);
         memset(c,,sizeof(c));
         for(int i=; i<=n; ++i)
         {
             add(a[i],);
             sum[i] = getSum(a[i]-);
         }
         memset(c,,sizeof(c));
         ll ans =;
         for(int i=n; i>; --i)
         {
             add(a[i],);
             int tmp = getSum(a[i]-);
             ans += (ll)sum[i]*(n-i-tmp) + (ll)(i--sum[i])*tmp;
         }
         cout<<ans<<endl;
     }
     return ;
 }

注意，如果想输出long long int 用printf（）函数WA了的话，就考虑一下cout吧····