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。
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
typedef long long ll;
int n;
struct {
int h; //身高
int idx;//上一次位置
ll m; //移动长度
} q[N], tmp[N];
void merge_sort(int l, int r) {
if (l >= r) {
return;
}
int mid = l + r >> 1;
merge_sort(l, mid), merge_sort(mid + 1, r);
int k = l, i = l, j = mid + 1;
while (i <= mid && j <= r) {
if (q[i].h <= q[j].h) {
tmp[k++] = q[i++];
} else {
tmp[k++] = q[j++];
}
}
while (i <= mid) {
tmp[k++] = q[i++];
}
while (j <= r) {
tmp[k++] = q[j++];
}
//更新上一位置 idx
//将本次移动长度累加到总长度
for (i = l; i < k; i++) {
q[i] = {tmp[i].h, i, tmp[i].m + abs(tmp[i].idx - i)};
}
}
int main() {
cin >> n;
for (int i = 1; i <= n; i++) {
cin >> q[i].h;
q[i].idx = i;
}
merge_sort(1, n);
ll cnt = 0;
for (int i = 1; i <= n; i++) {
cnt += (q[i].m + 1) * q[i].m / 2;//等差数列
}
cout << cnt << endl;
return 0;
}