农夫约翰的农场上有 N 个山丘,每座山的高度都是整数。
在冬天,约翰经常在这些山上举办滑雪训练营。
不幸的是,从明年开始,国家将实行一个关于滑雪场的新税法。
如果滑雪场的最高峰与最低峰的高度差大于17,国家就要收税。
为了避免纳税,约翰决定对这些山峰的高度进行修整。
已知,增加或减少一座山峰 x 单位的高度,需要花费 x的平方 的金钱。
约翰只愿意改变整数单位的高度。
请问,约翰最少需要花费多少钱,才能够使得最高峰与最低峰的高度差不大于17。
输入格式
第一行包含整数 N。
接下来 N 行,每行包含一个整数,表示一座山的高度。
输出格式
输出一个整数,表示最少花费的金钱。
数据范围
1≤N≤1000,
数据保证,每座山的初始高度都在 0∼100 之间。
输入样例:
5
20
4
1
24
21
输出样例:
18
样例解释
最佳方案为,将高度为 1 的山峰,增加 3 个单位高度,
将高度为 24 的山峰,减少3 个单位高度。
#include <iostream>
using namespace std;
const int N = 1010;
int n, h[N];
int main()
{
cin >> n;
for (int i = 0; i < n; ++i) cin >> h[i];//读取n个山峰高度
int res = 1e8;//花费最多不会超过100 * 100 * 1000
for (int i = 0; i <= 83; ++i)//枚举窗口底部所有可能的位置
{
int j = i + 17, cost = 0;//j:窗口顶部的值
for (int k = 0; k < n; ++k)//k:遍历所有山峰高度
//大于窗口口顶部产生的花费
if (h[k] > j) cost += (h[k] - j) * (h[k] - j);
//小于窗口底部产生的花费
else if (h[k] < i) cost += (i - h[k]) * (i - h[k]);
res = min(res, cost);//总花费取最小值
}
cout << res << endl;
return 0;
}