AcWing 1215. 小朋友排队
一、题目链接
AcWing 1215. 小朋友排队
二、题目分析
(一)算法标签
树状数组 归并排序 贪心
(二)解题思路
对于每一个数
,设在它前面
且比它大的数
的个数
为x
,在它后面
且比它小的数
的个数
为y
,则这个数共要交换n(其中, n = x + y)
次,其不满意度为
n
(
n
+
1
)
2
\frac {n(n + 1)} 2
2n(n+1)次
遍历加上每个数的不满意度得到的结果就是答案
三、AC代码
解法一:
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 1000010;
int n;
int tr[N], h[N];
int sum[N];
int lowbit(int x)
{
return x & -x;
}
void update(int x, int v)
{
for (int i = x; i < N; i += lowbit(i) ) tr[i] += v;
}
int query(int x)
{
int sum = 0;
for (int i = x; i; i -= lowbit(i)) sum += tr[i];
return sum;
}
int main()
{
cin >> n;
for (int i = 1; i <= n; i ++ )
{
scanf("%d", &h[i]);
h[i] ++ ;
sum[i] = query(N) - query(h[i]); // h[i]之前比h[i]大的数有多少个
update(h[i], 1);
}
memset(tr, 0, sizeof tr);
for (int i = n; i >= 1; i -- )
{
sum[i] += query(h[i] - 1); // h[i]之后比h[i]小的数有多少个
update(h[i], 1);
}
LL res = 0;
for (int i = 1; i <= n; i ++ ) res += (LL)sum[i] * (sum[i] + 1) / 2;
cout << res;
return 0;
}
四、其它题解
AcWing 1215. 小朋友排队