题目描述 Description
给定一个序列a1,a2,…,an,如果存在i<j并且ai>aj,那么我们称之为逆序对,求逆序对的数目
数据范围:N<=10^5。
Ai<=10^5。时间限制为1s。
输入描述 Input Description
第一行为n,表示序列长度,接下来的n行,第i+1行表示序列中的第i个数。
输出描述 Output Description
所有逆序对总数.
样例输入 Sample Input
4
3
2
3
2
样例输出 Sample Output
3
该题为典型的分治题目,其中利用了归并排序的思想
逆序对对数由两种组成
(1)数组左半部分和数组右半部分的逆序对对数之和
(2)一个数在左半部分,另一个数在右半部分的逆序对对数
题解将在代码中的注释中呈现
以下为代码
(萌新为保险起见都用了long long型)
#include<iostream>
#include<algorithm>
#define MAX 100002
using namespace std;
long long a[MAX],b[MAX],cnt=0;
void mergesortandcnt(long long a[],long long s,long long e,long long b[]);//利用递归函数,将大数组分解为小数组,进行排序,并计算逆序对的对数
void Merge(long long a[],long long s,long long m,long long e,long long b[]);//用来将排好序的两组数组,按顺序归并到一个数组里,此题解运用**从大到小**的顺序排列
int main()
{
long long n;
cin>>n;//输入元素个数
for(int i=0;i<n;i++)
cin>>a[i];//输入元素
mergesortandcnt(a,0,n-1,b);//计算
cout<<cnt<<endl;//输出结果
return 0;
}
void mergesortandcnt(long long a[],long long s,long long e,long long b[])
{
if(s<e)
{
long long m=s+(e-s)/2;//将数组分为前后两部分,不用(e+s)/2,防止e+s过大
mergesortandcnt(a,s,m,b);//递归前一部分
mergesortandcnt(a,m+1,e,b);//递归后一部分
long long i=s,j=m+1;//计算第二种情况,即一个数在左半部分,另一个数在右半部分的逆序对对数,
//分别从的第一个元素开始比较
while(i<=m&&j<=e)//界限为两个数组的右边界
{
if(a[i]<=a[j])//假如左边的元素小于或等于右边的元素,不符合条件,所以j++,比较下一个右边的数
j++;
else//假如左边的元素大于右边的元素,符合条件,则a[j]之后的元素都符合条件,因此不必再去比较之后的元素
{
i++;
cnt+=e-j+1;//不比较之后的元素直接加上之后的个数即可
}
}
Merge(a,s,m,e,b);//归并排好序的两个数组
}
}
void Merge(long long a[],long long s,long long m,long long e,long long b[])//归并排好序的两个数组
{
long long pb=0,p1=s,p2=m+1;
while(p1<=m&&p2<=e)//比较元素的大小,从大到小排序
{
if(a[p1]>a[p2])
b[pb++]=a[p1++];
else
b[pb++]=a[p2++];
}//p1,p2可能未到右边界,以下两个循环将剩下的元素放入临时数组中
while(p1<=m)
b[pb++]=a[p1++];
while(p2<=e)
b[pb++]=a[p2++];
for(int i=0;i<e-s+1;i++)
a[s+i]=b[i];//注意必须是s+i
}