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

题目描述

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

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

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

输入输出格式

输入格式：

 

第一行：一个整数n

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

 

输出格式：

 

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

枚举最小值到最大值，高于削去，低于补上，min比较答案

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>

#define N 1000+7
using namespace std;

inline int sqr(int x)
{
	return x*x;
}
int n,ma=0,mi=1000+7;
int aans=200000000+9,ans;
int a[N];
int main()
 {
 	scanf("%d",&n);
 	for(int i=1;i<=n;i++)
 	 {
 	 	scanf("%d",&a[i]);
 	 	ma=max(ma,a[i]);
 	 	mi=min(mi,a[i]);
	  }
	
	for(int j=mi;j<=ma;j++)
	 {
	 	ans=0;
	 	 for(int i=1;i<=n;i++)
	 	  {
	 	  	if(a[i]-17>j)ans+=sqr(a[i]-j-17);
	 	  	 else if(j>a[i])ans+=sqr(a[i]-j);
		   }
		aans=min(aans,ans);   
	   }  
	printf("%d",aans);   
 }

输入输出样例

输入样例#1： 复制

5
20
4
1
24
21

输出样例#1： 复制

18

说明

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