Acwing 算法基础课复习 逆序对的数量 巧用归并排序

---

题目链接

链接: 逆序对的数量

题目描述

解题思路

(yxc老师的图) 可知逆序对一共有三种形式我们只需要考虑黄色的即可，红色的和绿色的可以通过归并排序的递归调用过程算出来

如果a[i]>a[j],因为左右都已排好序，所以a[i]后面的元素都与a[j]构成逆序对，所以只需要在归并排序的基础上，当a[i]>a[j]时，让结果加上mid-i+1即可

归并排序：

1.[L, R] => [L, mid],[mid +1, R]
2．递归排序［L，mid］和［mid＋1，R］
3．归并，将左右两个有序序列合并成一个有序序列
归并排序模板如下

void merge_sort(int q[],int l,int r)
{
    if(l>=r) return;
    int mid=(l+r)>>1;
    merge_sort(q,l,mid),merge_sort(q,mid+1,r);
    int i=l,j=mid+1,k=0;
    while(i<=mid&&j<=r)
    {
        if(q[i]<=q[j]) tmp[k++]=q[i++];
        else tmp[k++]=q[j++];
    }
    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];
}

代码实现

#include<iostream>
#include<cstring>
#include<algorithm>
#define _for(i,a,b) for(int i=(a);i<(b);i++)
#define _rep(i,a,b) for(int i=(a);i<=(b);i++)
using namespace std;
typedef long long LL;
const int N=100001;
int a[N],tmp[N],n;
LL ans;
void merge_sort(int a[],int l,int r)
{
    if(l>=r) return;
    int mid=(l+r)>>1;
    merge_sort(a,l,mid),merge_sort(a,mid+1,r);
    int i=l,j=mid+1,k=0;
    while(i<=mid&&j<=r)
    {
        if(a[i]>a[j]){
          tmp[k++]=a[j++];  
          ans+=mid-i+1;
        } 
        else tmp[k++]=a[i++];
    }
    while(i<=mid) tmp[k++]=a[i++];
    while(j<=r) tmp[k++]=a[j++];
    for(int i=0,j=l;j<=r;i++,j++) a[j]=tmp[i];
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    cin>>n;
    _for(i,0,n) cin>>a[i];
    merge_sort(a,0,n-1);
    cout<<ans;
    return 0;
}

总结

巧用递归排序的性质和分治的思想解决计算逆序对数量的问题