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 =20004,N2 = 100005;
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 =0;
    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=1; i<=n; ++i)
            scanf("%d",&a[i]);
        memset(c,0,sizeof(c));
        for(int i=1; i<=n; ++i)
        {
            add(a[i],1);
            sum[i] = getSum(a[i]-1);
        }
        memset(c,0,sizeof(c));
        ll ans =0;
        for(int i=n; i>0; --i)
        {
            add(a[i],1);
            int tmp = getSum(a[i]-1);
            ans += (ll)sum[i]*(n-i-tmp) + (ll)(i-1-sum[i])*tmp;
        }
        cout<<ans<<endl;
    }
    return 0;
}

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