利用归并排序求逆序对的数量

最新推荐文章于 2023-03-08 19:01:09 发布

yypSandra

最新推荐文章于 2023-03-08 19:01:09 发布

阅读量107

点赞数

文章标签： 1024程序员节

本文链接：https://blog.csdn.net/qq_41734963/article/details/120939214

版权

问题描述

5 3 2 1 4
逆序对为[5,3],[5,2],[5,1],[5,4],[3,2],[3,1],[2,1]
逆序对数量是7

归并排序

在这里插入图片描述

解题思路

逆序对分成三大类：
两个数同时出现在Left，merge_sort(L,mid)
两个数同时出现在Right，merge_sort(mid+1,R)
一个数出现在Left一个数出现在Right，
在这里插入图片描述
sj=mid-i+1(j从mid+1到r,i从l到mid)

代码

#include<iostream>
using namespace std;
typedef long long LL;
const int N = 100010;
int n;
int q[N];// 存放待排序的数组
int tmp[N];// 存放归并排序的结果数组

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 = 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);
    return 0;
}