Day-17 滑雪场设计（枚举、贪心）

考点：枚举、贪心

AcWing 1353.滑雪场设计

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

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

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

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

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

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

约翰只愿意改变整数单位的高度。

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

输入格式

第一行包含整数 NN。

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

输出格式

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

数据范围

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

输入样例：

5
20
4
1
24
21

输出样例：

18

样例解释

最佳方案为，将高度为 11 的山峰，增加 33 个单位高度，将高度为 2424 的山峰，减少 33 个单位高度。

分析：

1.易知，最优解中，所有山峰的高度必然在给定的高度0~100之间（不会小于0，也不会大于100）

2.有1知，我们只需要枚举所有可能的区间即可。[0,17] [1,18] ... ... [83,100]共84个不同的区间

3.如果在最优解区间左边，就加上相应的高度，将其修整为最优解区间的左端点。如果在最优解区间右边，那么就减去相应高度，将其修整为最优解区间右端点。这样可以保证，修整的高度最小，花费最少。

4.看最终哪个区间的花费最少。

#include <iostream>

using namespace std;

const int N = 1010;

int n;
int h[N];

int main()
{
    cin >> n;
    for(int i = 0; i < n; i ++ ) cin >> h[i];
    
    int res = 1e8;
    for(int i = 0; i + 17 <= 100; i ++ ) //枚举最优解可能存在的每一个区间
    {
        int cost = 0, l = i, r = i + 17;
        for(int j = 0; j < n; j ++ )  
        {
            if(h[j] < l) cost += (l - h[j]) * (l - h[j]);
            else if(h[j] > r)
                cost += (h[j] - r) * (h[j] - r);
        }
        
        res = min(res,cost); //取能使花费最少的区间
    }
    
    cout << res;
    
    return 0;
}