蓝桥杯 PREV-31 小朋友排队

小朋友排队

时间限制：1.0s   内存限制：256.0MB

问题描述

　　n 个小朋友站成一排。现在要把他们按身高从低到高的顺序排列，但是每次只能交换位置相邻的两个小朋友。

每个小朋友都有一个不高兴的程度。开始的时候，所有小朋友的不高兴程度都是0。

如果某个小朋友第一次被要求交换，则他的不高兴程度增加1，如果第二次要求他交换，则他的不高兴程度增加2（即不高兴程度为3），依次类推。当要求某个小朋友第k次交换时，他的不高兴程度增加k。

请问，要让所有小朋友按从低到高排队，他们的不高兴程度之和最小是多少。

如果有两个小朋友身高一样，则他们谁站在谁前面是没有关系的。

输入格式

　　输入的第一行包含一个整数n，表示小朋友的个数。
第二行包含 n 个整数 H1 H2 … Hn，分别表示每个小朋友的身高。

输出格式

　　输出一行，包含一个整数，表示小朋友的不高兴程度和的最小值。

样例输入

3
3 2 1

样例输出

9

样例说明

　　首先交换身高为3和2的小朋友，再交换身高为3和1的小朋友，再交换身高为2和1的小朋友，每个小朋友的不高兴程度都是3，总和为9。

数据规模和约定

　　对于10%的数据， 1<=n<=10；
对于30%的数据， 1<=n<=1000；
对于50%的数据， 1<=n<=10000；
对于100%的数据，1<=n<=100000，0<=Hi<=1000000。

对于这样的排队问题，一般立刻想到的是逆序对数，但是这个题还有所提高在于他给每次的换位都增加了权值，导致不能一遍树状数组就能解决这个问题，对于任何一个位置的数，他的移动次数计算非常简单，其实就是它左面大于它的数加上它右面小于它的数，所以正向反向两次应用树状数组即可解决这个问题，这个题目的细节地方还是需要注意的，如题目范围给定小朋友高度为整数，所以会有0，所以在使用树状数组的时候需要按照对应身高+1进行，还有就是出现相同高度的情况，需要减去。

最后这个题目Hi的范围还是可以开数组的，不需要离散。

代码如下；

/*************************************************************************
	> File Name: PREV-31.cpp
	> Author: Zhanghaoran
	> Mail: chilumanxi@xiyoulinux.org
	> Created Time: 2016年02月19日 星期五 19时07分52秒
 ************************************************************************/

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cstdlib>

using namespace std;

long long tree[1000010];
long long int fri[1000010];
long long int step[1000010];
long long int inde[1000010];
long N;

void add(long long x){
    while(x < 100010){
        tree[x] ++;
        x += x & -x;
    }
}

long long check(long long x){
    long long sum = 0;
    while(x){
        sum += tree[x];
        x -= x & -x;
    }
    return sum;
}

int main(void){
    cin >> N;
    long long ans = 0;
    for(int i = 1; i <= N; i ++){
        inde[i] = inde[i - 1] + i;
    }
    for(int i = 1; i <= N; i ++){
        scanf("%lld", &fri[i]);
        add(fri[i] + 1);
        step[i] += i - check(fri[i]);
        step[i] -= check(fri[i] + 1) - check(fri[i]);
    }
    memset(tree, 0, sizeof(tree));
    for(int i = N; i >= 1; i --){
        add(fri[i] + 1);
        step[i] += check(fri[i]);
    }
    for(int i = 1; i <= N; i ++)
        ans += inde[step[i]];
    cout << ans << endl;
    
}

 

查看原文：http://chilumanxi.org/2016/02/19/%e8%93%9d%e6%a1%a5%e6%9d%af-prev-31-%e5%b0%8f%e6%9c%8b%e5%8f%8b%e6%8e%92%e9%98%9f/