1353. 滑雪场设计

https://www.acwing.com/problem/content/1355/

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

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

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

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

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

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

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

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

输入格式
第一行包含整数 N。

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

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

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

解题思路：枚举 + 贪心
枚举最优解的山峰所在区间，0-17 、 1-18、2-19、…
然后计算每个区间中不在区间内的山峰到区间端点值的差值和，取最小值，就是答案

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<string>
#include<iostream>
using namespace std;

int a[1005];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=0; i<n; ++i)
    {
        scanf("%d",&a[i]);
    }
    long long temp ,ans = 9999999;
    for(int i=0; i<=83; i++){
        temp = 0;
        int l = i,r = i+17;
        for(int j=0; j<n; ++j){
            if(a[j]<l){
                temp+= (l-a[j])*(l-a[j]);
            }
            else if(a[j]>r){
                temp += (a[j]-r)*(a[j]-r);
            }
        }
        ans = min(ans,temp);
    }
    printf("%lld\n",ans);
    return 0;
}