题目描述
农民约翰的农场里有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); }