【洛谷】P1428 小鱼比可爱

题目地址：

https://www.luogu.com.cn/problem/P1428

题目描述：
人比人，气死人；鱼比鱼，难死鱼。小鱼最近参加了一个“比可爱”比赛，比的是每只鱼的可爱程度。参赛的鱼被从左到右排成一排，头都朝向左边，然后每只鱼会得到一个整数数值，表示这只鱼的可爱程度，很显然整数越大，表示这只鱼越可爱，而且任意两只鱼的可爱程度可能一样。由于所有的鱼头都朝向左边，所以每只鱼只能看见在它左边的鱼的可爱程度，它们心里都在计算，在自己的眼力范围内有多少只鱼不如自己可爱呢。请你帮这些可爱但是鱼脑不够用的小鱼们计算一下。

输入格式：
第一行输入一个整数$n$，表示鱼的数目。
第二行内输入$n$个整数，用空格间隔，依次表示从左到右每只小鱼的可爱程度。

输出格式：
行内输出$n$个整数，用空格间隔，依次表示每只小鱼眼中有多少只鱼不如自己可爱。

数据范围：
对于$100\%$的数据，$n\le 100$。

思路是利用归并排序，将整个数组从小到大排序。在归并的时候，如果$a[i]\ge a[j]$，那么就知道了$a[l:i-1]<a[j]$，则可以将个数$i-l$累加给$a[j]$。由于这一题要对每个数求出其左边有多少个比它小的数，所以要对下标来排序。代码如下：

#include <iostream>
using namespace std;

const int N = 110;
int n;
int a[N], tmp[N], id[N];
int cnt[N];

void merge_sort(int l, int r) {
    if (l >= r) return;
    int mid = l + (r - l >> 1);
    merge_sort(l, mid); merge_sort(mid + 1, r);
    
    int i = l, j = mid + 1, idx = l;
    while (i <= mid && j <= r)
        if (a[id[i]] < a[id[j]]) tmp[idx++] = id[i++];
        else {
            cnt[id[j]] += i - l;
            tmp[idx++] = id[j++];
        }

    while (i <= mid) tmp[idx++] = id[i++];
    while (j <= r) {
        cnt[id[j]] += i - l;
        tmp[idx++] = id[j++];
    }

    for (int k = l; k <= r; k++) id[k] = tmp[k];
}

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        id[i] = i;
    }

    merge_sort(1, n);
    for (int i = 1; i <= n; i++) printf("%d ", cnt[i]);
    puts("");

    return 0;
}

时间复杂度$O(n\log n)$，空间$O(n)$。