13.1 数状数组 ——【小朋友排队】

题目描述

n个小朋友站成一排。现在要把他们按身高从低到高的顺序排列，但是每次只能交换位置相邻的两个小朋友。

每个小朋友都有一个不高兴的程度。开始的时候，所有小朋友的不高兴程度都是 0。

如果某个小朋友第一次被要求交换，则他的不高兴程度增加 1，如果第二次要求他交换，则他的不高兴程度增加 2（即不高兴程度为 3），依次类推。当要求某个小朋友第 k 次交换时，他的不高兴程度增加 k。

请问，要让所有小朋友按从低到高排队，他们的不高兴程度之和最小是多少。

如果有两个小朋友身高一样，则他们谁站在谁前面是没有关系的。

输入描述

输出描述

输出一行，包含一个整数，表示小朋友的不高兴程度和的最小值

输入输出样例

输入：

3
3 2 1

输出：

9

---

最终代码c/c++

#include <bits/stdc++.h>
using namespace std;
const int N = 1000010;
typedef long long LL;
int h[N], tree[N], k[N];        //h是身高，k是交换次数（逆序对数量）
int lowbit(int x) {  return x & -x; }
void update(int x, int d) {
    while(x <= N) {
        tree[x] += d;
        x += lowbit(x);
    }
}
int sum(int x) {
    int ans = 0;
    while(x > 0){
        ans += tree[x];
        x -= lowbit(x);
    }
    return ans;
}
int main() {
    int n; cin >> n;
    for (int i = 1; i <= n; i ++ ){
        cin >> h[i];
        h[i]++;          //h从1开始，不是从0开始
    }
    for (int i = 1; i <= n; i ++ ) {   //正序处理 逆序对，右边矮的
        k[i] = sum(N - 1) - sum(h[i]);
        update(h[i], 1);
    }
    memset(tree, 0, sizeof tree);
    for (int i = n; i; i-- ){         //倒序处理  逆序对，左边高的
        k[i] += sum(h[i] - 1);
        update(h[i], 1);
    }
    LL res = 0;
    for (int i = 1; i <= n; i ++ )    //最后把所有人的不愉快加起来
        res += (LL)k[i] * (k[i] + 1 ) / 2;
    cout << res;
    return 0;
}

---

过程理解