部分转自:https://blog.csdn.net/SSimpLe_Y/article/details/53744096
逆序对就是如果i > j && a[i] < a[j],这两个就算一对逆序对,简单来说,所有逆序对的个数和就是找每一个数的前面有几个比他的大的数,他们加起来的和就是逆序对的总数。
知道什么是逆序对后就好办了,树状数组的功能就是可以单点更新,区间查询,这样你把每个数该在的位置离散化出来,然后每次把每个数该在的位置上加上1,比如一个数该在的位置为x,那么用add(x)把这个位置加上1,然后再用区间查询read(x)查询1~x的和,也就是可以知道前面有多少个数是比他小的了(包括那个数自己),再用已经插入的数的个数减去read(x),就算出了前面有多少个数比他大了。
下面举个例子来详细的看一下过程:
第一次插入的时候把5这个位置上加上1,read(x)值就是1,当前已经插入了一个数,所以他前面比他大的数的个数就等于 i - read(x) = 1 - 1 = 0,所以总数 sum += 0
第二次插入的时候,read(x)的值同样是1,但是 i - read(x) = 2 - 1 = 1,所以sum += 1
第三次的时候,read(x)的值是2,i - read(x) = 3 - 2 = 1,所以sum += 1
第四次,read(x)的值是1,i - read(x) = 4 - 1 = 3,所以sum += 3
第五次,read(x)的值是1,i - read(x) = 5 - 1 = 4,所以sum += 4
这样整个过程就结束了,所有的逆序对就求出来了。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 10005
typedef long long ll;
int c[maxn];
int lowbit(int x)
{
return x&-x;
}
void update(int x,int val,int n)
{
for(int i=x;i<=n;i+=lowbit(i))
c[i]+=val;
}
int getsum(int x)
{
int ans=0;
for(int i=x;i;i-=lowbit(i))
ans+=c[i];
return ans;
}
struct node{
int val,pos;
}a[maxn];
int cmp(node a,node b)
{
return a.val<b.val;
}
int main()
{
int n;
int b[maxn];//存一个数应该在的位置
while(~scanf("%d",&n))
{
memset(c,0,sizeof(c));
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i].val);
a[i].pos=i;
}
sort(a+1,a+n+1,cmp);
//处理有数重复的情况
int cnt=1;
for(int i;i<=n;i++)
{
if(i!=1&&a[i].val!=a[i-1].val)
{
cnt++;
}
b[a[i].pos]=cnt;
}
ll sum=0;
for(int i=1;i<=n;i++)
{
update(b[i],1,n);
sum+=(i-getsum(b[i]));
}
cout<<sum<<endl;
}
return 0;
}