逆序对【洛谷P1908】

题目描述

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

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

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

输出格式：

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

输入输出样例
输入样例#1： 复制

6
5 4 2 6 3 1

输出样例#1： 复制

11

说明

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

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

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

请使用较快的输入输出

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

思路
每拿到一个值放入桶中，统计所有大于这个数的数量，然后累加
所以想到动态求和树状数组

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m,a[500005],b[500005];
long long c[500005];
void Add(int x)  
{
    for(;x<=n;x+=(x&-x))
        c[x]++; 
}
long long ask(int x)  
{
    long long  s=0; 
    for(;x>0;x-=(x&-x)) 
        s+=c[x];
    return s;  
}
bool cmp(const int &x,const int &y)  
{
    return b[x]>b[y];
}
int main()
{
    int i;  
    scanf("%d",&n);  
    for(i=1;i<=n;i++)
    {
        scanf("%d",&b[i]); 
        a[i]=i; 
    }
    sort(a+1,a+n+1,cmp); 
	long long ans=0; 
    for(i=1;i<=n;i++)
    {
        Add(a[i]); 
        ans+=ask(a[i]-1);
    }
    printf("%lld",ans); 
    return 0; 
}