POJ 3928 树状数组

这道题是08年北京现场赛的题。

讲一下题意，N个人，从中选出3个人，使得，这三个人必须是从大到小或者从小到大，并且是严格的大小关系，即不可相等。问有多少种这样的组合。

这道题的结论很好得出，假设一个人位于i，那么我们只需求出求1 -> (i - 1)位置小于这个人的个数，记为x ，当然大于这个人的个数也就出来了，记为x'，同理, 求出(i + 1) -> n位置小于这个人的个数，记为y ，那么大于他的个数即为y'。

那么最后的答案就是sigam(x * y' + x' * y) 。从1 -> n累加即可。

那么问题就转化成求一个数他前面有多少个数小于他，后面有多少个数小于他。

对于这个问题，我们显然可以用树状数组来求解，对于求前缀的时候，可以从1 -> n这样插入，然后每次插入都求一次前缀和即可。

同理对于求后面的个数，可以从n -> 1这样插，然后同样每次求前缀和即可。

当然因为必须是严格的大小关系，每次求和的时候必须是query(a[i] - 1)，这样就可以了。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <stack>
#include <map>
#include <iomanip>
#define PI acos(-1.0)
#define Max 2005
#define inf 2000000000
#define LL(x) (x<<1)
#define RR(x) (x<<1|1)
#define REP(i,s,t) for(int i=(s);i<=(t);++i)
#define ll long long
#define mem(a,b) memset(a,b,sizeof(a))
#define mp(a,b) make_pair(a,b)
using namespace std;

#define N 222222
int c[N] ;
int a[N] ;
int pre[N] ;
int suf[N] ;
int n , m ;
int lowbit(int x){return x & (-x) ;}

void updata(int x ){
    for (int i = x ; i <= m ; i += lowbit(i))c[i] ++ ;
}

int query(int x){
    int ans = 0 ;
    for (int i = x ; i >= 1 ; i -= lowbit(i)){
        ans += c[i] ;
    }
    return ans ;
}

void init(){
    mem(c ,0) ;
    mem(pre ,0) ;
    mem(suf, 0) ;
}
void solve(){
    cin >> n ;
    init() ;
    m = 0 ;
    for (int i = 1 ; i <= n ; i ++ ){
        scanf("%d",&a[i]) ;
        m = max(a[i] , m) ;
    }

    for (int i = 1 ; i <= n ; i ++ ){
        updata(a[i]) ;
        pre[i] = query(a[i] - 1) ;
    }
    mem(c ,0) ;
    for (int i = n ; i >= 1 ; i -- ){
        updata(a[i]) ;
        suf[i] = query(a[i] - 1) ;
    }
    ll ans = 0 ;
    for (int i = 1 ; i <= n ; i ++ ){
        ans += (ll)pre[i] * (n - i - suf[i]) + (ll)(i - pre[i] - 1) * suf[i] ;
    }
    cout << ans << endl;
}

int main() {
    int _ ;cin >> _ ;while(_ -- ) solve() ;
    return 0;
}