USACO 1.3 Ski Course Design(skidesign)

描述

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

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

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

格式

PROGRAM NAME: skidesign

INPUT FORMAT:

(file skidesign.in)

第一行：一个整数n

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

OUTPUT FORMAT:

(file skidesign.out)

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

SAMPLE INPUT

5
20
4
1
24
21

SAMPLE OUTPUT

18

HINTS

样例输入说明

约翰的农场有5座山,海拔为1,4,20,21和24

样例输出说明

约翰保持高度为4、20和21的山的高度。他增高高度为1的山、变成高度4(花费 3 ^ 2 = 9)。他降低了高度为24的山变成高度21也花费3 ^ 2 = 9。

题解：定义变量k为各个山峰需要调整到的高度（选取标准），然后从1至山峰高度的最大值进行枚举。

       如果当前高度小于标准，则ans+=差值*差值，如果当前高度大于标准且减去标准大于17，则ans+=(差值-17)*(差值-17）

#include<bits/stdc++.h>
using namespace std;
template<typename T>inline void read(T &x){
	int f=1;x=0;char c=getchar();
	for(;!isdigit(c);c=getchar())if(c=='-')f=-1;
	for(;isdigit(c);c=getchar())x=x*10+c-'0';
	x*=f;
}
int a[1010];
int main(){
	int n;
	cin>>n;
	
	int ma=0;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		ma=max(ma,a[i]);
	}
	
	int ans,result=INT_MAX;
	for(int i=1;i<=ma;i++){
		ans=0;
		for(int j=1;j<=n;j++){
			if(i>a[j]){
				ans+=(i-a[j])*(i-a[j]);
			}
			if(a[j]-i>17){
				ans+=(a[j]-i-17)*(a[j]-i-17);
			}
		}
		result=min(result,ans);
	}
	cout<<result<<endl;
	return 0;
}