[USACO1.3]滑雪课程设计Ski Course Design（洛谷P3650）

[USACO1.3]滑雪课程设计Ski Course Design

题目描述

农民约翰的农场里有N座山峰(1<=N<=1000),每座山都有一个在0到100之间的整数的海拔高度。在冬天,因为山上有丰富的积雪,约翰经常开办滑雪训练营。

不幸的是,约翰刚刚得知税法在滑雪训练营方面有新变化,明年开始实施。在仔细阅读法律后,他发现如果滑雪训练营的最高和最低的山峰海拔高度差大于17就要收税。因此,如果他改变山峰的高度（使最高与最低的山峰海拔高度差不超过17）,约翰可以避免支付税收。

如果改变一座山x单位的高度成本是x^2单位,约翰最少需要付多少钱?约翰只愿意改变整数单位的高度。

输入格式

第一行：一个整数n

第二行到N+1行：每行是一座山的海拔高度

输出格式

约翰需要支付修改山海拔高度的总金额,最高和最低的山峰间高度差最多17。

输入 #1

5
20
4
1
24
21

输出 #1

18

纯思维题，因为山峰高度为0到100，所以枚举0到83为改造后的最小山峰高度，然后根据题目要求计算就行；

代码：

#include<bits/stdc++.h>
using namespace std;
int a[1010];
int main(){
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	int mmin=1e9;
	for(int i=0;i<=83;i++){//枚举最低山峰的高度 
		int sum=0;
		for(int j=1;j<=n;j++){
			if(a[j]<i){
				sum+=(i-a[j])*(i-a[j]);	
			}
			else if(a[j]>i+17){
				sum+=(a[j]-i-17)*(a[j]-i-17);
			}
		}
		mmin=min(sum,mmin);
	}
	printf("%d\n",mmin);
	return 0;
}