2022年庐阳区青少年信息学科普日真题- 特别的电梯(lift)

题目描述
爸爸公司的大厦，在早高峰期，几乎每层都有人上下，电梯每层都停，大家被弄的很不耐烦，阳阳提出一个解决办法:电梯从一楼出发，只允许电梯停在其中的某一层，所有乘客再从该层走楼梯到自己的目的楼层。我们要给电梯编写一个附加特别程序，根据乘客在一楼按下的目标楼层，计算电梯应该停在哪一楼层，能够保证这次乘坐电梯的所有乘客爬楼梯的层数之和最少。求这个最小值。

输入格式
第一行一个正整数 N。

第二行，N 个整数，表示N 个人各自要去的楼层。

输出格式
一个整数，表示N 个人最小爬楼层数和。

输入输出样例

输入样例1：
5
10 3 8 7 4
输出样例1：
11

说明
数据范围

N<=100000

提示说明

爬楼梯上一层、下一层都记为1 次

---

【解析】
此题类似考察枚举；枚举电梯停在每一层，乘客走的楼层之和。

#include<bits/stdc++.h>
using namespace std;
int a[100010];
int b[10000100];
int minn=21e8;
int maxn=-21e8;
int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		maxn=max(maxn,a[i]); // 简单优化一下，求出最高楼层号
	}
	for(int i=1;i<=maxn;i++){ // 枚举 i 层为停靠点
		for(int j=1;j<=n;j++){
			b[i]+=abs(a[j]-i);
		}
	}
	for(int i=1;i<=maxn;i++){
		minn=min(minn,b[i]);
	}
	cout<<minn;
	return 0;
}

如果以上代码超时,可能各个系统出题者模拟的数据规模不同吧. 上面代码是通过了我们系统的.
以下附加一个优解吧.

#include<bits/stdc++.h> 
using namespace std;
int n,a[100100],ans;
int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}	
	sort(a,a+n);
	for(int i=0;i<n;i++){
		ans+=abs(a[n/2]-a[i]);
	}
	cout<<ans;
	return 0;
}