欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

求逆序对 - 归并排序

程序员文章站 2022-05-10 11:51:47
...

题目连接:http://ybt.ssoier.cn:8088/problem_show.php?pid=1311

n开到1e5的话,硬刚肯定是要超时的,所以可以用归并排序,并在排序的时候求出结果,其特点如下:

在每次合并两个子数组的时候,这两个子数组都已经是有序的啦

然后若a[i]>a[j]的话ans+=m-i+1

a[i]>a[j],则a[i-m]都>a[j]所以ans+=m-i+1

并且在比较两个数组元素的过程中,元素的相对位置不变,所以结果是正确的

代码如下:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

const int INF=0x3f3f3f3f;
const int N=100005;
int a[N],b[N],n;
ll ans=0;

void Merge(int s[],int t[],int b,int m,int e){
    int i=b,j=m+1,k=0;
    while(i<=m&&j<=e){
        if(s[i]>s[j]){
            ans+=m-i+1;
            t[k++]=s[j++];
        }
        else t[k++]=s[i++];
    }
    while(i<=m)t[k++]=s[i++];
    while(j<=e)t[k++]=s[j++];
    for(int q=0;q<k;q++){
        s[q+b]=t[q];
    }
}

void MSort(int s[],int t[],int b,int e){
    if(b<e){
        int m=(b+e)>>1;
        MSort(s,t,b,m);
        MSort(s,t,m+1,e);
        Merge(s,t,b,m,e);
    }
}

int main(){
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d",&a[i]);
    }
    MSort(a,b,0,n-1);
    printf("%lld\n",ans);
}

 

相关标签: 归并排序