hdu2492 Ping pong

hdu2492 Ping pong 

题意：一群乒乓爱好者居住在一条直线上，如果两个人想打比赛需要一个裁判，裁判的 位置 必须在两者之间 ，裁判的能力也必须不大于 参赛者最大的，不小于参赛者最小的

白皮的题解：考虑 第 i 个 为裁判 的情况 如果 左边 比 a[i] 小的 人数 为 c[i],则 有  i-c[i]-1 个人 比 a[i] 大，同样 右边 比 a[i] 小 的人数为 d[i] ，则比a[i]大的人为 n - i - d[i];

则方案 为 d[i]*( i - c[i] - 1 ) + c[i] * (n - i - d[i])

用 x[ a[i] ] 表示 比 a[i] 小的个数，这就涉及到了 树状数组

代码……

#include <iostream>

#include <cstring>

#include <cstdio>

#include <cmath>

#include <algorithm>

#include <vector>

#include <map>

#include <queue>

#include <stack>

#include <set>

#include <string>

using namespace std;

typedef long long ll;

const double ESP = 10e-8;

const int MOD = 1000000007;

typedef long long LL;

const int MAXN = 20000 + 10;

const int MAXA = 100000 + 10;

int bit[MAXA];

int c[MAXN];

int d[MAXN];

int arr[MAXN];



int sum(int i){

    int s = 0;

    while(i > 0){

        s += bit[i];

        i -= i&-i;

    }

    return s;

}



void add(int i,int x){

    while(i < MAXA){

        bit[i] += x;

        i += i&-i;

    }

}

int main(){

//    freopen("input.txt","r",stdin);

    int t;

    scanf("%d",&t);

    while(t--){

        int n;

        scanf("%d",&n);

        memset(bit,0,sizeof(bit));

        memset(c,0,sizeof(c));

        memset(d,0,sizeof(d));

        for(int i = 1;i <= n;i++){

            scanf("%d",&arr[i]);

            c[i] = sum(arr[i]);

            add(arr[i],1);

        }

        memset(bit,0,sizeof(bit));

        for(int i = n;i > 0;i--){

            d[i] = sum(arr[i]-1);

            add(arr[i],1);

        }

        LL ans = 0;

        for(int i = 1;i <= n;i++){

            ans += c[i]*(n-i-d[i]) + d[i]*(i-c[i]-1);

        }

        printf("%I64d\n",ans);

    }

    return 0;

}