逆序对（树状数组）

洛谷P1908-逆序对（树状数组）

一、题目描述

猫猫 TOM 和小老鼠 JERRY 最近又较量上了，但是毕竟都是成年人，他们已经不喜欢再玩那种你追我赶的游戏，现在他们喜欢玩统计。

最近，TOM 老猫查阅到一个人类称之为“逆序对”的东西，这东西是这样定义的：对于给定的一段正整数序列，逆序对就是序列中 $a_i>a_j$ 且 i<j 的有序对。知道这概念后，他们就比赛谁先算出给定的一段正整数序列中逆序对的数目。注意序列中可能有重复数字。

Update:数据已加强。

• 输入格式

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

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

• 输出格式

输出序列中逆序对的数目。

• 输入输出样例

**输入 **

6
5 4 2 6 3 1

**输出 **

11

• 说明/提示

对于 25% 的数据，n≤2500

对于 50% 的数据，n≤4×$10^4$。

对于所有数据，n≤5×$10^5$。

请使用较快的输入输出

应该不会 $O(n^2)$ 过 50 万吧 by chen_zhe

二、解析

求逆序对的数目我们可以使用树状数组或者归并排序，在这里我们给出树状数组的解法，之后会补上归并的解法（给自己挖了个坑！)。

• 建立树状数组st，初始值为零，树状数组中记录数据出现的次数。
• 我们依次遍历数据，每遍历一个数据，就将其所在位置的树状数组中的值加一（实现统计其所出现的次数），第i次加入数据时总的数据有i个，我们求出前i项和即比i小的元素的个数，用i减去比i小的元素的个数即得到此时比i大的元素的个数。树状数组中比其大的都会与其构成逆序对。
• 那么其逆序对的个数=比i大的元素的个数=i-getsum（$a_i$）($a_i$为第i个元素)。
• 我们举个例子来说明一下：以题目中数据为例，当遍历5时，逆序对数=1-1=0.(i=1,前五项和getsum(5)=1)遍历4时，逆序对数=2-1=1（i=2,前四项和=1），当遍历2时，逆序对数=3-1=2（i=3,前两项和=1），当遍历6时，逆序对数=4-4=0（i=4,前六项和=4），当遍历3时，逆序对数=5-2=3(i=5,前三项和=2)，当遍历1时，逆序对数=6-1=5（i=6,前一项和=1）.总的逆序数=0+1+2+0+3+5=11.
• 树状数组具体实现可看树状数组

三、代码实现

#include<iostream>
#include<algorithm>
using namespace std;
struct stu{
    int num,id;
}st[510000];
int n;
int tree[510000],tmp[510000];
bool cmp(stu a,stu b){
    if(a.num==b.num)
        return a.id<b.id;
    return a.num<b.num;
}//离散化
int lowbit(int x){
    return x&-x;
}
void add(int k,int x){
    while (k<=n)
    {
        tree[k]+=x;
        k+=lowbit(k);
    }
}
long long getsum(int k){
    long long ans=0;
    while (k)
    {
        ans+=tree[k];
        k-=lowbit(k);
    }
    return ans;
}

int main(){
    long long sum=0;
    cin>>n;
    int x;
    for(int i=1;i<=n;i++){
        scanf("%d",&st[i].num);
        st[i].id=i;
    }
    sort(st+1,st+n+1,cmp);
    for(int i=1;i<=n;i++){
        tmp[st[i].id]=i;
    }
    for(int i=1;i<=n;i++){
        add(tmp[i],1);
        sum+=i-getsum(tmp[i]);
    }
    printf("%lld",sum);
    system("pause");
    return 0;
}