蓝桥杯 小朋友排队

。。。一开始傻傻暴力的。。

后来才知道是树状数组求逆序数。。

还是先看清楚数据量啊！

#include<iostream>

#include<cstdio>

#include<cstdlib>

#include<cstring>

#include<string>

#include<queue>

#include<algorithm>

#include<map>

#include<iomanip>

#include<climits>

#include<string.h>

#include<cmath>

#include<stdlib.h>

#include<vector>

#include<stack>

#include<set>

using namespace std;

#define INF 1000000007

#define MAXN 4010

#define Mod 1000007

#define N 100010

#define NN 30

#define sigma_size 3

const int MAX = 1000100;

const int maxn = 6e5 + 10;

using namespace std;

typedef long long LL;





LL total[N], ans;

int num[N], C[MAX], S[MAX], b[N];

int lowbit(int x)

{

    return x&(-x);

}



void add(int pos,int num,int *p)

{

    while (pos <= MAX) {

        p[pos] += num;

        pos += lowbit(pos);

    }

}



int sum(int end, int *p)

{

    int cnt = 0;

    while (end > 0) {

        cnt += p[end];

        end -= lowbit(end);

    }

    return cnt;

}



void init()

{

    total[0] = 0;

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

        total[i] = total[i - 1] + i;

    }

}



int main()

{

    int t;

    init();

    while (cin >> t){

        for (int j = 0; j < t; ++j){//因为第一个数前面比它小的数没有，所以j要从0开始

            cin >> num[j];

            add(num[j] + 1, 1, C);

            b[j] = j - sum(num[j], C);;//Sum(num[j],C)求的就是小于num[j]的个数，j - Sum(num[j],C)就是前j个数中大于num[j]的个数  

            b[j] -= sum(num[j] + 1, C) - sum(num[j], C) - 1;//不包括本身这个数

        }

        ans = 0;

        for (int j = t - 1; j > -1; --j) {//反过来求第j个数右边中小于它的数的个数

            add(num[j] + 1, 1, S);

            b[j] += sum(num[j], S);//sum(num[j],S)求的就是小于num[j]的个数

            ans += total[b[j]];

        }

        cout << ans << endl;

    }

    return 0;

}