题目描述
农民约翰的农场里有N座山峰(1<=N<=1000),每座山都有一个在0到100之间的整数的海拔高度。在冬天,因为山上有丰富的积雪,约翰经常开办滑雪训练营。
不幸的是,约翰刚刚得知税法在滑雪训练营方面有新变化,明年开始实施。在仔细阅读法律后,他发现如果滑雪训练营的最高和最低的山峰海拔高度差大于17就要收税。因此,如果他改变山峰的高度(使最高与最低的山峰海拔高度差不超过17),约翰可以避免支付税收。
如果改变一座山x单位的高度成本是x^2单位,约翰最少需要付多少钱?约翰只愿意改变整数单位的高度。
输入格式
第一行:一个整数n
第二行到N+1行:每行是一座山的海拔高度
输出格式
约翰需要支付修改山海拔高度的总金额,最高和最低的山峰间高度差最多17。
输入输出样例
输入 #1复制
5 20 4 1 24 21
输出 #1复制
18
说明/提示
约翰保持高度为4、20和21的山的高度。他增高高度为1的山、变成高度4(花费 3 ^ 2 = 9)。他降低了高度为24的山变成高度21也花费3 ^ 2 = 9。
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
int a[1610],n,minn,ans=2147483647,sum;
int read(){
int a=0,b=1;
char ch=getchar();
while((ch<48||ch>57)&&ch!='-'){
ch=getchar();
}
if(ch=='-'){
b=-1;
ch=getchar();
}
while(ch<48||ch>57){
ch=getchar();
}
while(ch>47&&ch<58){
a=a*10+ch-48;
ch=getchar();
}
return a*b;
}
int main(){
n=read();
for(int i=1;i<=n;i++){
a[i]=read();
}
sort(a+1,a+n+1);
for(int j=a[1];j<=a[n];j++){
sum=0;
for(int i=1;i<=n;i++){
if(a[i]-j>17){
sum+=(a[i]-j-17)*(a[i]-j-17);
}
if(a[i]<j){
sum+=(a[i]-j)*(a[i]-j);
}
}
ans=min(sum,ans);
}
printf("%d",ans);
}