P1908 逆序对

题目描述

猫猫TOM和小老鼠JERRY最近又较量上了，但是毕竟都是成年人，他们已经不喜欢再玩那种你追我赶的游戏，现在他们喜欢玩统计。最近，TOM老猫查阅到一个人类称之为“逆序对”的东西，这东西是这样定义的：对于给定的一段正整数序列，逆序对就是序列中ai>aj且i<j的有序对。知道这概念后，他们就比赛谁先算出给定的一段正整数序列中逆序对的数目。
Update:数据已加强。

输入输出格式

输入格式：

 

第一行，一个数n，表示序列中有n个数。

第二行n个数，表示给定的序列。序列中每个数字不超过10^9109

 

输出格式：

 

给定序列中逆序对的数目。

 

输入输出样例

输入样例#1： 复制

6
5 4 2 6 3 1

输出样例#1： 复制

11

说明

对于25%的数据，n \leq 2500n≤2500

对于50%的数据，n \leq 4 \times 10^4n≤4×104。

对于所有数据，n \leq 5 \times 10^5n≤5×105

请使用较快的输入输出

应该不会n方过50万吧 by chen_zhe

思路：比较典型的树状数组的题目，注意需要离散化，然后求坐标后面比他小的数，在这里我们倒着输入即可

#include<iostream>
#include<algorithm>
using namespace std;
#define lowbit(i) ((i) & (-i))
const int maxn = 500010;
long long c[maxn], a[maxn];
long long n, k;
long long ans;
struct Node{
    long long val;
    int pos;
}node[maxn];

bool cmp(Node a, Node b) {
    return a.val < b.val;
}

int getSum(int x){
    int sum = 0;
    for (int i = x; i > 0; i -= lowbit(i)) {
        sum += c[i];
    }
    return sum;
}

void update(int x) {
    for (int i = x; i < maxn; i += lowbit(i)) {
        c[i] += 1;
    }
}

int main() {
    cin >> n;
    for(int i = 0; i < n; i++) {
        cin >> node[i].val;
        node[i].pos = i;
    }
    sort(node, node + n, cmp);
    for (int i = 0; i < n; i++) {
        if (i == 0 || node[i].val != node[i - 1].val) {
            a[node[i].pos] = i + 1;
        } else {
            a[node[i].pos] = a[node[i - 1].pos];
        }
    }
    for (int i = n - 1; i >= 0; i--) {
        update(a[i]);
        ans += getSum(a[i] - 1);
    }
    cout << ans << endl;
    return 0;
}