788.逆序对的数量

Wizard go

已于 2022-01-21 14:39:48 修改

阅读量364

点赞数

CC 4.0 BY-SA版权

分类专栏：算法刷题文章标签：排序算法算法数据结构

于 2022-01-21 14:39:06 首次发布

本文链接：https://blog.csdn.net/weixin_45774311/article/details/122620548

算法刷题专栏收录该内容

41 篇文章

订阅专栏

逆序对的数量

题目描述
给定一个序列有n个数，求n个数中逆序对的个数，逆序对的定义：i < j && a[i] > a[j]。
输入格式
第一行包含整数 n
第二行包含 n
输出格式
输出一个整数，表示逆序对的个数。
数据范围
1≤n≤100000
输入样例
6
2 3 4 5 6 1
输出样例
5

解法

本质还是归并排序，对于左右两段有序序列
在这里插入图片描述
如果 a[i] > a[j] ，那么i到mid之间的数都大于 a[j] ，则其中的逆序对数量为 mid-i+1

#include <iostream>
using namespace std;
typedef long long LL;
const int N = 100010;

int a[N], tmp[N];

LL merge_sort(int q[],int l,int r)
{
    if(l>=r) return 0;
    
    int mid=l+r>>1;
    
    LL res=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
        {
            res+=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];
    return res;
}

int main()
{
    int n;
    scanf("%d", &n);
    for (int i = 0; i < n; i ++ ) scanf("%d", &a[i]);
    cout<<merge_sort(a,0,n-1)<<endl;
    return 0;

}