Implementation：Binary Indexed Tree 树状数组

#include <iostream>
#include <cstdlib>

using namespace std;

class BinaryIndexedTree {
private:
    int *mem;
    int capacity;
public:
    BinaryIndexedTree (int n) {
        if (n <= ) {
            capacity = ;
            return;
        }
        capacity = n;
        mem = new int[capacity + ];
        for (int i=; i<=capacity; i++) mem[i] = ;
    }
    ~BinaryIndexedTree() {
        delete[] mem;
    }
    int sum(int idx) {
        if (idx++ >= capacity) idx = capacity;
        int s = ;
        while(idx > ) {
            s += mem[idx];
            idx -= idx & -idx;
        }
        return s;
    }

    void add(int idx, int val) {
        idx++;
        while (idx <= capacity) {
            mem[idx] += val;
            idx += idx & -idx;
        }
    }
};

int main() {
    int nums[] = {, , , , , , , };
    int n = sizeof(nums) / sizeof(int);

    BinaryIndexedTree bit(n);

    for (int i=; i<n; i++) {
        bit.add(i, nums[i]);
    }

    for (int i=; i<n; i++) {
        cout<<"["<<<<", "<<i<<"] :"<<bit.sum(i)<<endl;
    }

    // solve a problem using BIT, calculate how many reverse order pairs
    // in a shuffled array{1, 2, 3...n}; n = array.length
    int array[] = {,,,,};
    int size = sizeof(array) / sizeof(int);
    BinaryIndexedTree bt(size);
    int rsum = ;
    for (int i=; i<size; i++) {
        rsum += i - bt.sum(array[i]);
        bt.add(array[i], );
    }
    cout<<rsum<<endl;

    system("pause");
    return ;
}

参考：

　　挑战程序设计竞赛第二版p176