AcWing：1215. 小朋友排队

 描述

n 个小朋友站成一排。

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

每个小朋友都有一个不高兴的程度。

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

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

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

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

输入格式

输入的第一行包含一个整数 n，表示小朋友的个数。

第二行包含 n 个整数 H1,H2,…,Hn，分别表示每个小朋友的身高。

输出格式

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

数据范围

1≤n≤100000,
0≤Hi≤1000000

输入样例：

3
3 2 1

输出样例：

9

样例解释

首先交换身高为3和2的小朋友，再交换身高为3和1的小朋友，再交换身高为2和1的小朋友，每个小朋友的不高兴程度都是3，总和为9。

 对逆序对的理解和对树状数组的使用

以下是AC代码

/* 暴力 */
/*
#include <bits/stdc++.h>

using namespace std;

typedef long long LL;
const int N = 1e5 + 10;
LL h[N] , num[N];
LL n , ans = 0;

LL summ(LL x)
{
    LL k ;
    k = (1 + x) * x / 2;
    return k;
}

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

    for(int i = 1;i <= n;i ++)
    {
        LL res = 0;
        for(int j = 1;j <= i;j ++)
        {
            if(h[j] > h[i]) res ++;
        }

        for(int j = i + 1;j <= n;j ++)
        {
            if(h[j] < h[i]) res ++;
        }

        res = summ(res);
        num[i] =res;
    } 
    
    for(int i = 1;i <= n; i ++) ans += num[i];

    cout << ans;
    return 0;
}

*/

/* 树状数组 */

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;
const int N = 1e6 + 10;
int n;
LL h[N], tree[N], b[N] = {0};

/* 最低位1的位数，固定函数 */
int lowbit(int x)
{
    return x & -x;
}

/* 添加树状数组函数 ，保存的不是数，而是该数的出现次数 */
void add(int x, int y)
{
    for(int i = x;i < N; i += lowbit(i)) tree[i] += y;
}
/* 查询前面比该数小的数 */
int query(int x)
{
    int ans = 0;
    for(int i = x;i ; i -= lowbit(i)) ans += tree[i];
    return ans;
}
int main()
{
    /* 读入 */
    cin >> n;
    for(int i = 1;i <= n; i ++) 
    {
        cin >> h[i];
        h[i] ++; // 避免0的情况
    }

    /* 添加到树状数组中 */
    for(int i = 1;i <= n;i ++)
    {
        add(h[i], 1);
        b[i] = (i) - query(h[i]); // 统计比该数大的数
    }

    /* 逆序对大的算完，要清零先 */
    memset(tree , 0 , sizeof tree);

    /* 倒序添加 */
    for(int i = n;i >= 1;i --)
    {
        add(h[i] ,1);
        b[i] += query(h[i] - 1); // 比该数小的
    }

    LL res = 0;
    for(int i = 1;i <= n;i ++) res += (1 + b[i]) * b[i] / 2;

    cout << res;
    return 0;
}