acwing 1353. 滑雪场设计

1353. 滑雪场设计

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

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

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

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

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

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

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

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

输入格式

第一行包含整数 NN。

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

输出格式

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

数据范围

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

输入样例：

5
20
4
1
24
21

输出样例：

18

样例解释

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

 

思路：因为只要求最高峰与最低峰高度差小于等17，而且数据范围较小，可以枚举每一对(x, x + 17)，然后比较每个需要花费的值，取最小值即可

#include <bits/stdc++.h>
using namespace std;
const int N =  1e5 + 10;
int n, q[N], res = INT_MAX;

int main()
{
    cin >> n;
    for (int i = 0; i < n; i ++) cin >> q[i];
    for (int i = 1; i < 100 - 17 + 1; i ++)
    {
        int l = i, r = i + 17, tem = 0;
        for (int j = 0; j < n; j ++)
        {
            if (q[j] < l) tem += pow(l - q[j], 2);
            else if (q[j] > r) tem += pow(q[j] - r, 2);
        }
        res = min(res, tem);
    }
    cout << res;
    return 0;
}