描述
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;
}