nyist oj 117 求逆序数 (归并排序&&树状数组）

求逆序数

时间限制：2000 ms  |  内存限制：65535 KB
难度：5

描述

在一个排列中，如果一对数的前后位置与大小顺序相反，即前面的数大于后面的数，那么它们就称为一个逆序。一个排列中逆序的总数就称为这个排列的逆序数。

现在，给你一个N个元素的序列，请你判断出它的逆序数是多少。

比如 1 3 2 的逆序数就是1。

输入第一行输入一个整数T表示测试数据的组数（1<=T<=5)

每组测试数据的每一行是一个整数N表示数列中共有N个元素（2〈=N〈=1000000）

随后的一行共有N个整数Ai(0<=Ai<1000000000)，表示数列中的所有元素。

数据保证在多组测试数据中，多于10万个数的测试数据最多只有一组。
输出输出该数列的逆序数
样例输入

2
2
1 1
3
1 3 2

样例输出

0
1

来源[张云聪]原创
上传者张云聪

求逆序数：用归并排序来做，效率还是很高的；
归并排序利用分治的思想，先把一个数组分成一个个序列，然后对一个个序列排序，把排好序的序列，在合并到原来的数组中。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1000001;
int a[maxn],b[maxn];
long long sum;
void Merge(int begin,int mid,int end)//归并
{
int i=begin,j=mid+1,pos=begin;
while(i<=mid && j<=end)//对一个个序列排序的过程
{
if(a[i]<=a[j])
{
b[pos++]=a[i++];
}
else
{
b[pos++]=a[j++];
sum+=mid-i+1;//求逆序数
}
}
while(i<=mid) b[pos++]=a[i++];
while(j<=end) b[pos++]=a[j++];
for(int i=begin,j=begin;i<=end;i++,j++)
a[i]=b[j];
}
void Sort(int begin,int end)//排序
{
if(begin<end)
{
int mid=(begin+end)/2;
Sort(begin,mid);
Sort(mid+1,end);
Merge(begin,mid,end);
}
}
int main()
{
int t,n;
scanf("%d",&t);
while(t--)
{
sum=0;
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
Sort(1,n);
printf("%lld\n",sum);
}
return 0;
}

用树状数组求逆序数：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=1000005;
struct node
{
int val,id;
}s[maxn];
int a[maxn],n;
int lowbit(int i)//位运算
{
return i&(-i);
}
void update(int i)//更新
{
while(i<=n)
{
a[i]++;
i+=lowbit(i);
}
}
int sum(int i)//求和
{
int sum=0;
while(i>0)
{
sum+=a[i];
i-=lowbit(i);
}
return sum;
}
bool cmp(node x,node y)//比较函数
{
if(x.val!=y.val)
return x.val<y.val;
return x.id<y.id;
}
int main()
{
int t,i;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
for(i=1;i<=n;i++)
{
scanf("%d",&s[i].val);
s[i].id=i;
a[i]=0;
}
long long ans=0;
sort(s+1,s+n+1,cmp);//离散化
for(i=1;i<=n;i++)
{
update(s[i].id);
ans+=i-sum(s[i].id);
}
printf("%lld\n",ans);
}
return 0;
}