LA 4329 pingpong 树状数组

LA 4329 pingpong

P197

树状数组统计，统计数组中，一个值的左边比它小的个数

用乘法原理求方法总数。

注意

　　N个数互不相同(distinct)

　　因为，最大值有限。可以利用树状数组统计。 

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>

#define lowbit(x) ((x)&(-(x)))
using namespace std;

const int max_a=100000+1000;
const int max_num=20000+200;

typedef long long LL;

int a[20000+200];
LL cc[20000+200],dd[20000+200];
int n=max_a,C[max_a];

int sum(int x) {
    int ret = 0;
    while ( x>0 ) {
        ret += C[x];
        x -= lowbit(x);
    }
    return ret;
}

void add(int x, int d) {
    while(x <= n) {
        C[x] += d;
        x += lowbit(x);
    }
}

int main()
{
    //freopen("input.txt","r",stdin);
    int T;cin>>T;
    int num;
    for(int tt=1;tt<=T;tt++) {
        scanf("%d",&num);
        for(int i=0;i<num;i++)
            scanf("%d",&a[i]);


        memset(C,0,sizeof(C));
        for(int i=0;i<num;i++) {
            add(a[i],1);
            cc[i]=sum(a[i])-1;//cout<<cc[i]<<endl;
        }
        memset(C,0,sizeof(C));
        for(int i=num-1;i>=0;i--) {
            add(a[i],1);
            dd[i]=sum(a[i])-1;//cout<<cc[i]<<endl;
        }
        long long ans=0;
        for(int i=1;i<num-1;i++) {
            ans+=(cc[i])*(num-(1+i)-dd[i])+
                 (i-cc[i])*(dd[i]);
        }
        cout<<ans<<endl;
    }
    return 0;
}

//from http://code.google.com/p/aoapc-book/source/browse/TrainingGuide/bookcodes/ch3/la4329.cpp?spec=svn16d535c9811a6d2b26c1fbb2ba40fce4359a4d01&r=16d535c9811a6d2b26c1fbb2ba40fce4359a4d01

inline int lowbit(int x) { return x&-x; }

struct FenwickTree {
  int n;
  vector<int> C;

  void resize(int n) { this->n = n; C.resize(n); }
  void clear() { fill(C.begin(), C.end(), 0); }

  // ¼ÆËãA[1]+A[2]+...+A[x] (x<=n)
  int sum(int x) {
    int ret = 0;
    while(x > 0) {
      ret += C[x]; x -= lowbit(x);
    }
    return ret;
  }

  // A[x] += d (1<=x<=n)
  void add(int x, int d) {
    while(x <= n) {
      C[x] += d; x += lowbit(x);
    }
  }
};

BIT(Binary Indexed Tree)又名 Fenwick Tree

二叉搜索树可以完成这个任务吗？eg use STL/python的set

插入节点，返回比它小的数的个数。