HNU 程序设计训练 4.4 逆序数

问题描述

在一个排列中，如果一对数的前后位置与大小顺序相反，即前面的数大于后面的数，那么它们就称为一个逆序。一个排列中逆序的总数就称为这个排列的逆序数。也就是说，对于n个不同的元素，先规定各元素之间有一个标准次序（例如n个 不同的自然数，可规定从小到大为标准次序），于是在这n个元素的任一排列中，当某两个元素的先后次序与标准次序不同时，就说有1个逆序。一个排列中所有逆序总数叫做这个排列的逆序数。

比如：

数列 1 7 3 5 4 8 9

其中（7,3），（7,5），（7,4），（5,4）构成逆序，所以其逆序数为4。

对给定的数列，求出其逆序数。

输入格式

有多组测试数据。

每组测试数据第一行是一个正整数N，表示数列中元素个数，接下来一行N个用空格分隔开的正整数，表示数列的N个元素，数列元素值小于32768，并且一个数列中没有两个数值相同。

N=0表示输入结束，并且不需要处理。

40%的数列元素个数N 1 ≤ N≤ 10；

30%的数列元素个数N 1 ≤ N≤ 100；

20%的数列元素个数N 1 ≤ N≤ 1000；

10%的数列元素个数N 1 ≤ N≤ 5000；

输出格式

　　对于每组测试数据，输出一个整数：数列的逆序数。

样例输入

7
1 7 3 5 4 8 9
4
1 2 3 4
0

样例输出

4
0

// clang-format off
#include <bits/stdc++.h>
using namespace std;
using i64 = long long;
#define add emplace_back
#define arr(...) vector<__VA_ARGS__>
#define arrin(a) for (auto& i : a) cin >> i
#define multi_test int i; cin >> i; while (i--)
// clang-format on
const int inf = INT_MAX / 2;
const int mod = 998244353;
const int mxn = 5e5 + 10;
int test_case = 1;

int a[mxn], b[mxn];
i64 ans;

void merge_sort(int l, int r)
{
    if (l == r) {
        return;
    }
    int mid = (l + r) / 2;
    merge_sort(l, mid);
    merge_sort(mid + 1, r);
    int lp = l, rp = mid + 1;
    for (int i = l; i <= r; i++) {
        if (lp <= mid and (a[lp] <= a[rp] or rp > r)) {
            b[i] = a[lp++];
            ans += rp - mid - 1;
        } else {
            b[i] = a[rp++];
        }
    }
    for (int i = l; i <= r; i++) {
        a[i] = b[i];
    }
}

void solve()
{
    // cerr << "\ntest " << test_case++ << ":\n";
    while (true) {
        int n; cin >> n;
        if (n == 0) {
            return;
        }
        for (int i = 1; i <= n; i++) {
            cin >> a[i];
        }
        merge_sort(1, n);
        cout << ans << '\n';
        ans = 0;
    }
}

// clang-format off
int main()
{
    cin.tie(nullptr)->sync_with_stdio(false);
    //multi_test
    solve();
    return 0;
}