归并排序解决逆序对数问题

给定一个长度为n的整数数列,请你计算数列中的逆序对的数量。

逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i < j 且 a[i] > a[j],则其为一个逆序对;否则不是。

输入格式
第一行包含整数n,表示数列的长度。

第二行包含 n 个整数,表示整个数列。

输出格式
输出一个整数,表示逆序对的个数。

数据范围
1≤n≤100000
输入样例:

6
2 3 4 5 6 1

输出样例:

5

C++代码如下:

#include<iostream>
using namespace std;
#define N 100050

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


int main()
{
    
    
    int q[N];
    int n;
    cin >> n;
    for(int i = 0 ; i < n ; i ++) cin >> q[i];
    merge_sort(q,0,n-1);
    cout << sum;
    
    return 0;
}

猜你喜欢

转载自blog.csdn.net/diviner_s/article/details/107317196
今日推荐