农夫约翰的农场上有 N 个山丘,每座山的高度都是整数。
在冬天,约翰经常在这些山上举办滑雪训练营。
不幸的是,从明年开始,国家将实行一个关于滑雪场的新税法。
如果滑雪场的最高峰与最低峰的高度差大于17,国家就要收税。
为了避免纳税,约翰决定对这些山峰的高度进行修整。
已知,增加或减少一座山峰 x 单位的高度,需要花费 x^2 的金钱。
约翰只愿意改变整数单位的高度。
请问,约翰最少需要花费多少钱,才能够使得最高峰与最低峰的高度差不大于17。
输入格式
第一行包含整数 N。
接下来 N 行,每行包含一个整数,表示一座山的高度。
输出格式
输出一个整数,表示最少花费的金钱。
数据范围
1≤N≤1000,
数据保证,每座山的初始高度都在 0∼100 之间。
输入样例:
5
20
4
1
24
21
输出样例:
18
样例解释
最佳方案为,将高度为 1 的山峰,增加 3 个单位高度,将高度为 24 的山峰,减少 3 个单位高度。
思路
首先可以证明,最终这n棵树的长度不会大于100或者小于0。因为如果存在这样的高度,由于之前所有的高度都是在0-100区间里的,所以,这些这些不在范围内的高度是可以修建在0-100内的。而且这样的花费比在0-100范围外更少。
所以最终这n棵树的高度必定在一个小于等于17的区间内,由于所有高度不超过0-100,所以一共有【0-17】,【1-18】,【2-19】,…[83-17]这84个区间。我们只需要枚举这84个区间,在枚举每个区间的时候,如果某棵树不在这个区间内(大于或者小于),我们就使得这颗树修剪到离其高度最近的端点值即可
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
ll arr[1001];
int main(void)
{
int n;
cin >> n;
ll maxx = 0x3fffffff;
for (int i = 0; i < n; i++)cin >> arr[i];
for (int i = 0; i <= 83; i++)//枚举84个区间
{
ll sum = 0;
for (int j = 0; j < n; j++)
{
if (arr[j] > 17 + i) {//如果比右端点大,就修剪到右端点
sum += (arr[j] - i-17) * (arr[j] - i-17);
}
if (arr[j] < i) {//如果比左端点大,就修剪到左端点
sum += (i - arr[j]) * (i - arr[j]);
}
}
maxx = min(maxx, sum);//求修建到这n个区间的最小值
}
cout << maxx << endl;
return 0;
}