蓝桥杯历届试题 小朋友排队（树状数组，逆序对）

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

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

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

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

如果有两个小朋友身高一样，则他们谁站在谁前面是没有关系的。
输入格式
　　输入的第一行包含一个整数n，表示小朋友的个数。
　　第二行包含 n 个整数 H1 H2 … Hn，分别表示每个小朋友的身高。
输出格式
　　输出一行，包含一个整数，表示小朋友的不高兴程度和的最小值。
样例输入
3
3 2 1
样例输出
9
样例说明
　　首先交换身高为3和2的小朋友，再交换身高为3和1的小朋友，再交换身高为2和1的小朋友，每个小朋友的不高兴程度都是3，总和为9。
数据规模和约定
　　对于10%的数据， 1<=n<=10；
　　对于30%的数据， 1<=n<=1000；
　　对于50%的数据， 1<=n<=10000；
　　对于100%的数据，1<=n<=100000，0<=Hi<=1000000。

刚读完题给我的感觉就是有点像冒泡排序的过程，于是沿着这个思路我就想到了逆序对，然后看数据量发现暴力肯定是不行，幸好之前做过用树状数组求解逆序对，然后就完美的卡住了。。。。
首先这个题不是普通的求解逆序对，因为移动两个小朋友都会使得双方增加不高兴值，我们就必须要得到每个小朋友移动的次数。一个小朋友移动的原因肯定是他前面有比他大的或者他后面有比他小的，顺着这个思路我们就可以在输入的时候找到他前面比他大的。然后再倒着遍历一下找比他小的。这样也就求得了以当前位置为基准得到的逆序对的数量，也就是当前位置移动的次数。

第一次暴力代码得了60分：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

int arr[100005],sum[100005];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    scanf("%d",&arr[i]);

    for(int i=1;i<=n;i++)
    {
        for(int j=i+1;j<=n;j++)
        if(arr[i] > arr[j])
        ++sum[i],++sum[j];
    }
    long long res = 0;
    for(int i=1;i<=n;i++)
    res += (sum[i]+1)*sum[i]/2;

    cout<<res<<endl;
    return 0;
}

AC代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
#define lowbit(x) x&(-x)
#define maxn 100005
const int maxh = 1000005; //身高最大值

//height记录学生身高，树状数组使用cnt，num用来记录逆序对的数量
int height[maxn],cnt[maxh],num[maxn];
long long arr[maxn];  //等差前n项和打表，移动n次就是1~n的和
void init()
{
    arr[1] = 1;
    for(int i=2;i<maxn;i++)
    arr[i] = arr[i-1]+i;
}
int n;
void add(int x,int y)
{
    while(x <= maxh)
    {
        cnt[x] += y;
        x += lowbit(x);
    }
}
int sum(int x)
{
    int ans = 0;
    while(x)
    {
        ans += cnt[x];
        x -= lowbit(x);
    }
    return ans;
}
int main()
{
    init();   //打表等差数列
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&height[i]);
        height[i]++;  //防止出现0
        //找比当前身高高的，先add在sum
        add(height[i],1);
        //找到小于等于当前身高的，再用i减去得到的结果就是他前面比他高的人数
        num[i] = i-sum(height[i]);
    }
    //倒着找，找比他矮的
    memset(cnt,0,sizeof(cnt));
    for(int i=n;i>=1;i--)
    {
    	//先sum在add
        num[i] += sum(height[i]-1);
        add(height[i],1);
    }
    
    long long res = 0;
    for(int i=1;i<=n;i++)
    res += arr[num[i]];
    cout<<res<<endl;
    return 0;
}