https://www.acwing.com/problem/content/1355/
农夫约翰的农场上有 N 个山峰,每座山的高度都是整数。
在冬天,约翰经常在这些山上举办滑雪训练营。
不幸的是,从明年开始,国家将实行一个关于滑雪场的新税法。
如果滑雪场的最高峰与最低峰的高度差大于17,国家就要收税。
为了避免纳税,约翰决定对这些山峰的高度进行修整。
已知,增加或减少一座山峰 x 单位的高度,需要花费 x^2 的金钱。
约翰只愿意改变整数单位的高度,且每座山峰只能修改一次。
请问,约翰最少需要花费多少钱,才能够使得最高峰与最低峰的高度差不大于17。
输入格式
第一行包含整数 N。
接下来 N 行,每行包含一个整数,表示一座山的高度。
输出格式
输出一个整数,表示最少花费的金钱。
数据范围
1≤N≤1000,
数据保证,每座山的初始高度都在 0∼100 之间。
解题思路:枚举 + 贪心
枚举最优解的山峰所在区间,0-17 、 1-18、2-19、…
然后计算每个区间中不在区间内的山峰到区间端点值的差值和,取最小值,就是答案
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<string>
#include<iostream>
using namespace std;
int a[1005];
int main()
{
int n;
scanf("%d",&n);
for(int i=0; i<n; ++i)
{
scanf("%d",&a[i]);
}
long long temp ,ans = 9999999;
for(int i=0; i<=83; i++){
temp = 0;
int l = i,r = i+17;
for(int j=0; j<n; ++j){
if(a[j]<l){
temp+= (l-a[j])*(l-a[j]);
}
else if(a[j]>r){
temp += (a[j]-r)*(a[j]-r);
}
}
ans = min(ans,temp);
}
printf("%lld\n",ans);
return 0;
}