AcWing788.逆序对的数量

最新推荐文章于 2024-07-08 22:12:26 发布

CE_Chicken

最新推荐文章于 2024-07-08 22:12:26 发布

阅读量76

点赞数 1

分类专栏：算法基础课文章标签：算法 c++

本文链接：https://blog.csdn.net/canying6/article/details/127137336

版权

算法基础课专栏收录该内容

6 篇文章 0 订阅

订阅专栏

题目描述

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

逆序对的定义如下：对于数列的第 $i$ 个和第 $j$ 个元素，如果满足 $i < j$ 且 $a\left[i\right]\ >\ a\left[j\right]$ ，则其为一个逆序对；否则不是。

输入格式

第一行包含整数 $n$ ，表示数列的长度。

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

输出格式

输出一个整数，表示逆序对的个数。

数据范围

$1\ \le\ n\ \le\ 100000$ ，
数列中的元素的取值范围 $1\ \sim\ 10^{9}$ 。

输入样例：

6
2 3 4 5 6 1

输出样例：

图解（图来自Hasity大佬）

在这里插入图片描述
所以，这样我们就可以用一个变量res来记录逆序对总数
并将原来归并排序的函数改为 $\ long$ 类型，然后将一个大区间分为两半，将两半统计的逆序对数量相加即可。

My Code

#include <iostream>

using namespace std;

typedef long long LL;

const int N = 1e5 + 10;

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
        {
            tmp[k ++ ] = q[j ++ ];
            res += mid - i + 1;
        }
    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;
}