AcWing寒假每日一题——Day17滑雪场设计

1353滑雪场设计

一、问题描述

农夫约翰的农场上有 $N$ 个山丘，每座山的高度都是整数。在冬天，约翰经常在这些山上举办滑雪训练营。不幸的是，从明年开始，国家将实行一个关于滑雪场的新税法。如果滑雪场的最高峰与最低峰的高度差大于17，国家就要收税。为了避免纳税，约翰决定对这些山峰的高度进行修整。已知，增加或减少一座山峰 $x$ 单位的高度，需要花费 $x^2$ 的金钱。约翰只愿意改变整数单位的高度。请问，约翰最少需要花费多少钱，才能够使得最高峰与最低峰的高度差不大于17。
输入格式
第一行包含整数 $N$ 。接下来 $N$ 行，每行包含一个整数，表示一座山的高度。
输出格式
输出一个整数，表示最少花费的金钱。
数据范围
1≤$N$≤1000,数据保证，每座山的初始高度都在 0∼100 之间。
输入样例：

5
20
4
1
24
21

输出样例：

18

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

二、问题分析

由题目知，最终山峰所在的高度区间只可能在$[0,100]$之间，我们只需要将$[0,100]$之中的83个长度为17的区间枚举一遍找到哪个区间的所花钱数最少即可。
代码如下（示例）：

#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int N,q[1010];
int main()
{
    cin>>N;
    for(int i=0;i<N;i++)cin>>q[i];
    int res=1e9;
    for(int i=0;i<=83;i++)
    {
        int sum=0,l=i,r=i+17;
        for(int j=0;j<N;j++)
        {
            if(q[j]<l)sum+=pow((l-q[j]),2);//在区间左边的加到右边界
            else if(q[j]>r)sum+=pow((q[j]-r),2);//在区间右边的加到左边界
        }
        res=min(sum,res);
    }
    cout<<res<<endl;
    return 0;
}