初见蓝桥——历年试题 小朋友排队

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/fanyuwgy/article/details/74978925

蓝桥杯——历年试题 小朋友排队

问题描述

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。

 

 

这题真是搞死我了,开始老是往冒泡排序上面想,因为排序常用熟悉买,发现不能解决不高兴程度最小的问题。

后来想到了逆序数(这个读者可以自己百度一下),最先想的当然是暴力,看看数据范围后我果断放弃了。想到了树状数组(读者可以自己百度)。下面我来详细分析一下。

我们以3 2 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

可以看到sum(C[1],C[4])=1(可以由树状数组的统计数组得到),这个是小于等于3的数字的个数,也就是说当输入第一个数字3的时候没有比它小的数字存在,这时我们用 输入数字总数-sum(C[1],C[4])=0,也就是说大于3的数字的个数为0,我们令b[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

可以看到sum(C[1],C[3])=1,任然不存在比它小的数,但此时输入的数据总量为2,而2-1=1,就是说,存在一个数在2之前并且大于2,这个数当然就是3,我们另b[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

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

 

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

之后就是求和而已了,代码如下:

#include<iostream>
#include<string.h>
#define MAX 1000010
#define N 100010 
using namespace std;
int C1[MAX],C2[MAX],b[MAX];
int num[N];
long long total[N],ans;
int lowbit(int x)
{
	return x&(-x);
}

void add(int pos,int num,int *C)
{
	while(pos<=MAX)
	{
		C[pos]+=num;
		pos+=lowbit(pos);
	}
}

int Sum(int pos,int *C)
{
	int sum=0;
	while(pos>0)
	{
		sum+=C[pos];
		int a=lowbit(pos);
		pos-=lowbit(pos);
//		cout<<"yufan:"<<pos<<" "<<a<<endl;
	}
	return sum;
}

void init()
{
	total[0]=0;
	for(int i=0; i<=N; i++)
	{
		total[i] = total[i-1]+i;
	}
}
int main()
{
	int i=0;
	int n=0;
	init();
	memset(C1,0,sizeof(C1));
	memset(C2,0,sizeof(C2));
	cin>>n;
	for(i=0; i<n; i++)
	{
		cin>>num[i];
		add(num[i]+1,1,C1);
		b[i]=i-Sum(num[i],C1);
		cout<<Sum(num[i],C1)<<endl;
		b[i]=b[i]-(Sum(num[i]+1,C1)-Sum(num[i],C1)-1);
	}
	for(i=n-1;i>=0;i--)
	{
		add(num[i]+1,1,C2);
		b[i]=b[i]+Sum(num[i],C2);
	}
	ans=0;
	for(i=0;i<n;i++)
	{
		ans+=total[b[i]];
	}
	cout<<ans;
	return 0;
} 


猜你喜欢

转载自blog.csdn.net/fanyuwgy/article/details/74978925