P8613 [蓝桥杯 2014 省 B] 小朋友排队 题解

复习蓝桥杯的时候碰到了这道题，发现题解有的并没有讲得很清楚，所以想在这里补充一下。

首先我们看一下有关题意：

        简而言之就是给你一个乱序序列，之后让你求最少付出多少代价就可以使得整个序列为有序序列，我们最开始想到的一定是归并排序，很好，这说明我们对这个问题是有所了解的，但是，因为归并排序只是简单求逆序对数量，我们无法直到最终每个小朋友的交换次数，所以这种方法是行不通的。

        那么现在问题的核心在于如何快速求每个小朋友对应的逆序对数量（前面比他高的与后面比他矮的数量之和），遇事不决，肯定可以暴力去做，但是暴力是双重for循环，肯定会超时，所以有没有一种快速求得某个区间中的比某个数字大的个数的方法呢，这时我们可以使用树状数组。

        树状数组适用于求某个区间和，同时可以动态的进行添加，我们结合这两个性质思考一下，由于所有的小朋友数量是固定的，我们可以按照下面的步骤求小朋友数量：

1、首先按顺序读入，每次读入一个，就查询前面比这个小朋友高的数量，之后将这个小朋友的身高添加进去

2、之后逆序读入，每次读入一个，就查询后面比这个小朋友身高矮的数量，之后将这个小朋友的身高添加进去

        按照这样的步骤我们就可以求出每个小朋友的交换次数，之后求1+2+3+...+n即可，这里可以用等差数列性质求解。

下面附上源代码：

#include<iostream>
using namespace std;
#include<cstring>

typedef long long LL;

//本题每个小朋友交换的次数就是每个小朋友前后逆序对的数量
//所以结果就是求出每个小朋友逆序对的数量并进行累加
//但是问题核心就在于如何求出每个小朋友对应的逆序对数量
const int N = 1e6+10;
int tr[N],sum[N],h[N];
//tr为每个小朋友前面比他身高高的人的数量或者后面比他身高矮的数量，会进行实时添加
int n;

//下面三个为树状数组常用函数
int lowbit(int x) {
    return x&-x;
}

void add(int u,int v) {
    for(int i=u;i<N;i+=lowbit(i)) tr[i]+=v;
}

int query(int x){
    int res=0;
    for(int i=x;i>0;i-=lowbit(i)) res+=tr[i];
    return res;
}

int main() {
    cin>>n;
    
    //还有身高为0的小朋友
    for(int i=1;i<=n;i++) scanf("%d",&h[i]),h[i]++;
    
    //先统计前面比他身高高的小朋友数量
    for(int i=1;i<=n;i++){
        //小于最大身高数量-当前小朋友身高数量
        sum[i]+=query(N-1)-query(h[i]);
        add(h[i],1);
    }
    
    memset(tr,0,sizeof(tr));
    //之后统计后面比他身高矮的小朋友
    //这里要清空tr数组
    for(int i=n;i>=1;i--){
        sum[i]+=query(h[i]-1);
        add(h[i],1);
    }
    
    LL res=0;
    //一共交换sum[i]次，等差数列求和公式
    for(int i=1;i<=n;i++) res+=(LL)sum[i]*(sum[i]+1)/2;
    
    printf("%lld",res);
    
    return 0;
}

以上就是全部内容了，通过这道题我们可以总结经验，树状数组不仅仅能够实现区间查询与实时修改，同时还可以求逆序对数量（实际上也是区间查询与实时修改的变种）