0
点赞
收藏
分享

微信扫一扫

AcWing 1215. 小朋友排队 【树状数组】

扶摇_hyber 2022-03-22 阅读 85

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. 小朋友排队

举报

相关推荐

0 条评论