AcWing 1215. 小朋友排队 【第五届蓝桥杯省赛C++B/C组】【树状数组】

---

一、题目链接

AcWing 1215. 小朋友排队

---

二、题目分析

（一）算法标签

树状数组 归并排序 贪心

（二）解题思路

对于每一个数，设在它前面且比它大的数的个数为x，在它后面且比它小的数的个数为y，则这个数共要交换n(其中, n = x + y)次，其不满意度为$\frac{n(n+1)}{2}$次
遍历加上每个数的不满意度得到的结果就是答案
暴力做法：$O(n^2)$ (会TLE)

// 统计第x个数之前且比h[x]大的数的个数
int cal_cnt1(int x)
{
    int sum = 0;
    for (int i = 1; i < x; i ++ )
        if (h[i] > h[x]) sum ++ ;
    return sum;
}

// 统计第x个数之后且比h[x]小的数的个数
int cal_cnt2(int x)
{
    int sum = 0;
    for (int i = n; i > x; i -- )
        if (h[i] < h[x]) sum ++ ;
    return sum;
}

main():

for (int i = 1; i <= n; i ++ )
{
    scanf("%d", &h[i]);
    sum[i] = cal_cnt1(i);    // h[i]之前比h[i]大的数有多少个
}

for (int i = n; i >= 1; i -- )
{
    sum[i] += cal_cnt2(i);  // h[i]之后比h[i]小的数有多少个
}

树状数组做法：$O(nlogn)$

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

main():

for (int i = 1; i <= n; i ++ )
{
    scanf("%d", &h[i]);
    h[i] ++ ;   // 加1使下标从1开始(h[i]可能为0)
    sum[i] = query(N) - query(h[i]);    // h[i]之前比h[i]大的数有多少个
    update(h[i], 1);
}
// 重置tr数组
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);
}

---

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