给定一个长度为 n 的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i<j 且 a[i]>a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n,表示数列的长度。
第二行包含 n 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤100000,
数列中的元素的取值范围 [1,109]。
输入样例:
6
2 3 4 5 6 1
输出样例:
5
上代码:
#include<iostream>
using namespace std;
const int N=100010;
int q[N];
int tem[N];
long long int merge_sort(int q[],int l,int r)
{
if (l>=r) return 0;
int mid = l+r>>1;
int i=l,j=mid+1,k=0;
long long num = merge_sort(q,l,mid)+merge_sort(q,mid+1,r);
while(i<=mid && j<=r)
if(q[i]<=q[j]) tem[k++]=q[i++];
else
{
tem[k++]=q[j++];
num += mid-i+1;
}
while(i<=mid) tem[k++]=q[i++];
while(j<=r) tem[k++]=q[j++];
for(i=l,j=0;i<=r;i++,j++)
q[i]=tem[j];
return num;
}
int main()
{
int n;
cin>>n;
for(int i=0;i<n;i++)
cin>>q[i];
cout<<merge_sort(q,0,n-1);
}
首先我们分析一下这道题,题目让我们求逆序对的数目,那么什么是逆序对呢?逆序对就是在前面的数大于在后面的数,第一眼看到这道题的时候我是想用循环暴力求解,可是我们要注意这个数列元素数据范围是1≤n≤100000,时间复杂度太大。
这段代码采用了归并算法,在上一篇文章中我讲到了归并算法是先把一个大的数组拆成各自独立的个体再进行排序进行合并,相当于将问题变成了无数的两个数组值的比较,那我们要寻找逆序对就可以在这个排序的过程中加上逆序对数目。
我们仍然是按照举例子的方式对算法进行诠释,如下有5个数,让我们一起来按照算法寻找一下这其中的逆序对数。
首先,设置mid=1+r>>1,即取这五个数的中间数,min=0+4/2=2,q[2]=2,我们可以将数组分成左右两边。
在数组的不断划分递归下,最后会被划分成一个一个的数字,这时我们进行排序并且开始计数。
进行双指针计数,此时我们合并5与8,由于q[i]<q[j],所有tem[0]=5,由于j=l,tem[1]=8,此时num=0,第一个归并结束,返回num=0,将tem的值返回给q数组,由于在不断递归中的l和r是针对每一次递归的过程是不一样的,在返回q时按照的是最初的l和r,所以最后q所得到的数组一定是排好序的。
我们按照顺序将这些独立的小组再归并回去,在上图归并的过程中由于q[j]<q[i],所以tem[0]=q[j]=8,此时num=mid-i+1=1-0+1=2,代表在这次合并中有两对逆序对,实际上我们可以看出确实是两对逆序对,最后将i不断移动直到i>mid,将q[i]的数放入tem数组,返回num=0+2=2。
此时总数组的左边数已经归并完毕,我们再来归并右边的数。
由于j指向的数小于i指向的数,于是先将9填入tem数组,同时num=mid-i+1即0-0+1=1,表明有一对逆序对,注意这里的mid是两个独立数组上一层的mid,即两个数组还没有被分开时的mid,再填入14,返回num=1.
最后两个数组进行归并
返回num=2+1=3
我们再来看看5 8 2 14 9 这五个数,其中的逆序对有5 2 ,8 2 ,14 9 ,的确是三对,完美运算。