给定一个长度为 n� 的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i� 个和第 j� 个元素,如果满足 i<j 且 a[i]>a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数 n�,表示数列的长度。
第二行包含 n� 个整数,表示整个数列。
这道题暴力解决两层循环肯定是TLE,所以这里我们用归并排序的方法解决。对于左区间,右区间同时有序时,左区间(left,mid),右区间(mid+1,right).用i,j'分别指向left,mid+1;从小到大存到tmp数组里,如果到q[i]>q[j]的时候,即存右区间的数时,逆序对数量为mid-i+1
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=1e6+10;
int n,q[maxn],tmp[maxn];
LL merge_sort(int l,int r)
{
if(l>=r) return 0;
int mid= l+r>> 1;
LL res = merge_sort(l,mid)+ merge_sort(mid+1,r);
int k=0,i=l,j=mid+1;
while(i<=mid && j<=r)
{
if(q[i]<=q[j]) tmp[k++]=q[i++];
else
{
tmp[k++]=q[j++];
res+=mid-i+1;
}
}
while(i<=mid) tmp[k++]=q[i++];
while(j<=r) tmp[k++]=q[j++];
for(int i=l,j=0;i<=r;i++,j++) q[i]=tmp[j];
return res;
}
int main()
{
cin>>n;
for(int i=0;i<n;i++) cin>>q[i];
cout<<merge_sort(0,n-1)<<endl;
return 0;
}