USACO - Chapter1 Section 1.3 - Ski Course Design

Ski Course Design

题目描述

农民约翰的农场里有N座山峰(1<=N<=1000),每座山都有一个在0到100之间的整数的海拔高度。在冬天,因为山上有丰富的积雪,约翰经常开办滑雪训练营。

不幸的是,约翰刚刚得知税法在滑雪训练营方面有新变化,明年开始实施。在仔细阅读法律后,他发现如果滑雪训练营的最高和最低的山峰海拔高度差大于17就要收税。因此,如果他改变山峰的高度（使最高与最低的山峰海拔高度差不超过17）,约翰可以避免支付税收。

如果改变一座山x单位的高度成本是x^2单位,约翰最少需要付多少钱?约翰只愿意改变整数单位的高度。

输入输出格式

输入格式：
第一行：一个整数n

第二行到N+1行：每行是一座山的海拔高度

输出格式：
约翰需要支付修改山海拔高度的总金额,最高和最低的山峰间高度差最多17。

输入输出样例

输入样例#1：
5
20
4
1
24
21
输出样例#1：
18
说明

约翰保持高度为4、20和21的山的高度。他增高高度为1的山、变成高度4(花费 3 ^ 2 = 9)。他降低了高度为24的山变成高度21也花费3 ^ 2 = 9。

数据规模小，直接枚举每一个17范围的区间，再枚举每一座山的高度，如果不在区间内就把他调整一下，累计到 sum 里面，最终答案就是最小的一个 sum.

Code

#include<iostream>
#include<cmath>
using namespace std;
int main()
{
    int n,ans = 0x3f3f3f3f;
    cin >> n;
    int a[n];
    for (int i=0;i<n;i++) cin >> a[i];
    for (int i=0;i<=83;i++)
    {
        int sum = 0;
        for (int j=0;j<n;j++)
        {
            int temp = 0;
            if (a[j] < i) temp = i - a[j];
            if (i+17 < a[j]) temp = a[j]-i-17;
            if (temp != 0) sum += temp*temp;
        }
        ans = min(sum,ans);
    }
    cout << ans;
    return 0;
}