蓝桥杯 小朋友排队(树状数组+逆序数)

小朋友排队  
时间限制: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。


优质题解:http://www.aiuxian.com/article/p-2288507.html


这是一道寻找逆序数的题目,比如题目中的3,2,1,和3有关的逆序对为3,2和3,1,和2有关的逆序对为3,2和2,1,和1有关的逆序对为3,1和2,1,为了完成排序,任何一个逆序对的两个元素都必须交换一次,于是可知,每个小朋友都完成了两次交换。

说白了就是针对每一个元素,看看它之前有几个大于它的元素,之后有几个小于它的元素,然后记录这个元素在逆序对中出现的次数,也就是含这个元素的逆序对的个数,然后再将每个元素的不高兴程度相加即可。

问题的关键在于查找逆序对的个数,最容易想到的当然就是暴力大法,看看数据范围,超时无疑,于是我们就需要使用一个特殊的数组----树状数组。

首先我们看看树状数组的特点,当然就是很快速的求前n个元素的和,那这和我们求逆序对有什么关系呢?

这里,我们要巧妙地变一下形,还是以题目中给出的数据为例  3   2   1

因为树状数组实际上是由两部分组成:数据数组+统计数组,我们只看数据数组

由于树状数组是从1开始的,而题目中小盆友的身高可以为0(真是长见识了),所以我们将每个小盆友的身高加1然后作为树状数组的下标,将数值1存到相应的位置.

第一次读入3,此时读入的数据量为1个,变成这样

C[1]        C[2]        C[3]        C[4]       C[5]        C[6]         C[7]         C[8]
  0             0            0            1            0             0             0              0
可以看到getsum(4)=1(可以由树状数组的统计数组得到),这个是小于等于3的数字的个数,也就是说当输入第一个数字3的时候没有比它小的数字存在,这时我们用 输入数字总数-getsum(4)=0,也就是说大于3的数字的个数为0,我们令cou[0]=0.

第二次读入2,此时读入的数据量为2个,变成这样

C[1]        C[2]        C[3]        C[4]       C[5]        C[6]         C[7]         C[8]
  0             0            1            1            0             0             0              0

可以看到getsum(3)=1,任然不存在比它小的数,但此时输入的数据总量为2,而2-1=1,就是说,存在一个数在2之前并且大于2,这个数当然就是3,我们另cou[1]=1.

第三次读入1,此时读入的数据量为3,变成这样

C[1]        C[2]        C[3]        C[4]       C[5]        C[6]         C[7]         C[8]
  0             1            1            1            0             0             0              0
可以看到getsum(2)=1,任然不存在比它小的数,但此时输入的数据总量为3,而3-1=2,就是说,存在两个数在1之前并且大于1,这个数当然就是2,3,我们另cou[2]=2. 

到此,我们已经算出了每个数前面的较大的数的个数了,数据存在num[]中,现在我们再反过来,先插入1,再插入2,再插入3,但这次我们不再用总数减去getsum了,而直接求getsum,求出来的自然就是,每个数后面的较小的数的个数,然后将得到的数值累加到相应的cou[i]中,最终我们会得到cou[0]=2,cou[1]=2,cou[2]=2,分别对应num[0]=3,num[1]=2,num[2]=1。

求得了每个小盆友被移动的次数,我们需要计算其不高兴程度,这里实际上可以事先打个表,就是将被移动n次后的不高兴值全算出来,然后直接用就可以了,这里,我们将其存到total[]数组中,而且total[2]=3,所以总不高兴值就是9.


需要注意的是如果重复的数字出现怎么办,如果出现,实际上出问题的会是求每个数之前较大数的那部分,因为用到了数的总个数,如果出现一样的数,就会导致相减后的结果偏大,而且正好是大了 重复量-1 ,那么我们就可以算出重复量,然后将这一部分减去就行,关键是怎么算重复量,实际也很简单,通过树状数组,我们可以求得getsum(a)和sum(a+1),其中输入的数字为a,前者算出的小于a的数的个数,后者算出的是小于等于a的数的个数,两个一减就是等于a的个数。

#include<iostream>
#include<cstring>
using namespace std;
const int maxn = 1000010;
const int N = 100010;
int c1[maxn], c2[maxn], cou[N];
int num[N];
long long total[N];
void init()
{
    total[0] = 0;
    for(int i = 1; i <= 100000; i++)
        total[i] = total[i-1] + i;
}
int lowbit(int x)
{
    return x & (-x);
}
void add(int pos, int val, int *c)
{
    while(pos <= 1000005)
    {
        c[pos] += val;
        pos += lowbit(pos);
    }
}
int getsum(int pos, int *c)
{
    int sum = 0;
    while(pos > 0)
    {
        sum += c[pos];
        pos -= lowbit(pos);
    }
    return sum;
}
int main()
{
    init();
    int n;
    long long ans;
    while(cin >> n)
    {
        memset(c1, 0, sizeof(c1));
        memset(c2, 0, sizeof(c2));
        for(int i = 0; i < n; i++)
        {
            cin >> num[i];
            add(num[i]+1, 1, c1);
            cou[i] = i - getsum(num[i], c1);
            cou[i] = cou[i] - (getsum(num[i]+1, c1) - getsum(num[i], c1) - 1);
        }
        for(int i = n-1; i >= 0; i--)
        {
            add(num[i]+1, 1, c2);
            cou[i] += getsum(num[i], c2);
        }
        ans = 0;
        for(int i = 0; i < n; i++)
            ans += total[cou[i]];
        cout << ans << endl;
    }
    return 0;
}






  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值