線段樹 HDU 4000

題目鏈接：http://acm.hdu.edu.cn/showproblem.php?pid=4000

代碼風格：www.notonlysuccess.com

題目大意：求滿足下列條件數的對數：i < j < k && ai < ak < aj

算法：線段樹

思路：求大於ai 的數的個數ni，sum = ∑( ni * (ni-1) / 2) ),；再減去類似如下情況的數 1,3,4（即三個數從小到大排列的數的個數），方法見http://blog.csdn.net/zssee33/article/details/7925529。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;

#define lson l, m, rt << 1
#define rson m+1, r, rt << 1 | 1
#define mid int m = (l+r) >> 1

int p[456456], cnt[456456];
void update(int k, int l , int r, int rt)
{
    cnt[rt] ++;
    if(l == r)
    {
        return ;
    }
    mid ;
    if(k <= m)
        update(k, lson);
    else update(k, rson);
}

int query(int k, int l, int r, int rt)
{
    if(l == r)
        return cnt[rt];
    mid ;
    if(k <= m)
        return query(k, lson);
    else return cnt[rt << 1] + query(k, rson);
}

int main()
{
    int T, n;
    scanf("%d", &T);
    int ica = 1;
    while(T --)
    {
        scanf("%d", &n);
        printf("Case #%d: ", ica ++);
        int i;
        memset(cnt, 0, sizeof(cnt));
        __int64 sum = 0;
        for(i = 1; i <= n; i ++)
        {
            scanf("%d", &p[i]);
            __int64 k = query(p[i], 1, n, 1);
            update(p[i], 1, n, 1);
            __int64 b = n - p[i] - i + 1 + k;
            sum -= k * b;
            sum += (b) * (b-1) / 2;
        }
        printf("%I64d\n", sum % 100000007);
    }
    return 0;
}