1311：【例2.5】求逆序对

【题目描述】

给定一个序列a1,a2,…,an，如果存在i<j并且ai>aj，那么我们称之为逆序对，求逆序对的数目。

【输入】

第一行为n,表示序列长度，接下来的n行，第i+1行表示序列中的第i个数。

【输出】

所有逆序对总数。

【输入样例】

4

3

2

3

2

【输出样例】

3

【提示】

N≤105，Ai≤105。

分析

此题目利用归并排序的性质解题。归并排序可以参考——归并排序

在每次合并时，我们直接把满足逆序对的情况记录下来即可。

代码

#include<bits/stdc++.h>
using namespace std;

const int N=5e5+10;

int n;
int a[N],temp[N];
long long  ans=0; 

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 k=0,i=l,j=mid+1;
    while(i<=mid&&j<=r){
        if(a[i]>a[j]){
            ans=ans+mid-i+1;
            temp[k++]=a[j++];
        }
        else {
            temp[k++]=a[i++];
        }
    }
    while(i<=mid) temp[k++]=a[i++];
    while(j<=r) temp[k++]=a[j++];
    for(int i=l,j=0;i<=r;i++,j++){
        a[i]=temp[j];
    }
}

int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
    } 
    merge_sort(a,0,n-1);
//    for(int i=0;i<n;i++){
//        cout<<a[i]<<" ";
//    }
    cout<<ans<<endl;
}