求逆序对 - 归并排序

题目连接: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);
}

猜你喜欢

转载自blog.csdn.net/m0_37579232/article/details/84844226