1353. 滑雪场设计 枚举+贪心

1353. 滑雪场设计 枚举+贪心

农夫约翰的农场上有 N个山峰，每座山的高度都是整数。

在冬天，约翰经常在这些山上举办滑雪训练营。

不幸的是，从明年开始，国家将实行一个关于滑雪场的新税法。

如果滑雪场的最高峰与最低峰的高度差大于17，国家就要收税。

为了避免纳税，约翰决定对这些山峰的高度进行修整。

已知，增加或减少一座山峰 x单位的高度，需要花费 x2的金钱。

约翰只愿意改变整数单位的高度，且每座山峰只能修改一次。

请问，约翰最少需要花费多少钱，才能够使得最高峰与最低峰的高度差不大于17。

输入格式

第一行包含整数 N。

接下来 N行，每行包含一个整数，表示一座山的高度。

输出格式

输出一个整数，表示最少花费的金钱。

数据范围

1≤N≤1000,数据保证，每座山的初始高度都在 0∼100之间。

输入样例：

5
20
4
1
24
21

输出样例：

18

样例解释

最佳方案为，将高度为 1的山峰，增加 3 个单位高度，将高度为 24 的山峰，减少 3 个单位高度。

数据量为 1 ~ 1000，所以可以穷举每一种情况。找出最佳方案。

用 l 表示整改后最低的高度，则 l 的范围为 0 ~ 100-17。

遍历 l 的各个情况，算出每种情况的花费，花费最小的即为答案。

#include <iostream>
using namespace std;
const int N = 1010;
int a[N];
int main()
{int n;cin >> n;for(int i = 0; i < n; i++){cin >> a[i];}int res = 21E8;//最少花费，初值设置一个较大的值int l = 0;for(int l = 0; l <= 100 - 17; l++)//穷举每一种情况{int money = 0;//该方案的总花费for(int i = 0; i < n; i++){if(a[i] < l)//比 l 低的山峰需要整改，计算花费money += (a[i] - l) * (a[i] - l);else if(a[i] > l + 17) //比 l + 17 高的山峰需要整改，计算花费money += (a[i] -  l - 17) * (a[i] -  l - 17);}if(money < res)//如果该方案花费小于前面方案，则更新答案res = money;}cout << res;
}