题目描述:
n个小朋友站成一排。
现在要把他们按身高从低到高的顺序排列,但是每次只能交换位置相邻的两个小朋友。
每个小朋友都有一个不高兴的程度。
开始的时候,所有小朋友的不高兴程度都是 0。
如果某个小朋友第一次被要求交换,则他的不高兴程度增加 1,如果第二次要求他交换,则他的不高兴程度增加 2(即不高兴程度为 3),依次类推。当要求某个小朋友第 k 次交换时,他的不高兴程度增加 k。
请问,要让所有小朋友按从低到高排队,他们的不高兴程度之和最小是多少。
如果有两个小朋友身高一样,则他们谁站在谁前面是没有关系的。
输入格式:
输入的第一行包含一个整数 n,表示小朋友的个数。
第二行包含 n 个整数 H1,H2,…,Hn,分别表示每个小朋友的身高。
输出格式:
输出一行,包含一个整数,表示小朋友的不高兴程度和的最小值。
数据范围:
1≤n≤100000
0≤Hi≤1000000
输入样例:
3
3 2 1
输出样例:
9
题目分析:
我们很容易发现,小朋友交换时只能交换位置相邻的两个小朋友。这种方式与冒泡排序类似,我们可以采用相似的方法求解。但是,冒泡排序显然不是最好的方法,他的时间复杂度高达O(n^2),在本题中,无疑会超时。
由于小朋友交换时只能交换位置相邻的两个小朋友,故应当交换的两个小朋友实质上是一个逆序对。则在题目中交换的次数即为逆序对的数量。对每个小朋友,我们记在这个小朋友之前有k1个小朋友比他高,在他之后有k2个小朋友比他矮,易得,这个小朋友至少需要交换的次数为k1 + k2。于是乎,本题就转化为统计每个小朋友前比其升高搞得人数以及在其之后比其升高低的人数。此处用树状数组实现。
树状数组中,下标表示小朋友升高,值表示在一段身高范围内人数个数。
代码实现:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 1000010;
int h[N], tr[N], sum[N];
int n;
int lowbit(int x){
return x & -x;
}
void add(int x, int v){
for(int i = x; i < N; i += lowbit(i)) tr[i] += v;
}
int query(int x){
int res = 0;
for(int i = x; i; i -= lowbit(i)) res += tr[i];
return res;
}
int main(){
cin >> n;
for(int i = 0; i < n; i ++ ) scanf("%d", &h[i]), h[i] ++ ;
//首先计算每个数前面有多少个数比他大
for(int i = 0; i < n; i ++ ){
sum[i] = query(N - 1) - query(h[i]);
add(h[i], 1);
}
memset(tr, 0, sizeof(tr));
//计算每个数后面有多少的数比他小
for(int i = n - 1; i >= 0; i -- ){
sum[i] += query(h[i] - 1);
add(h[i], 1);
}
long long res = 0;
for(int i = 0; i < n; i ++ )
res += (long long) sum[i] * (sum[i] + 1) / 2;
cout << res << endl;
return 0;
}