洛谷 P3650 [USACO1.3]滑雪课程设计Ski Course Design

P3650 [USACO1.3]滑雪课程设计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。

标程

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,sum,ans=0x7f7f7f7f;
int num[1010];
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&num[i]);
    sort(num+1,num+1+n);
    for(int j=num[1];j<=num[n];j++){
        sum=0;
        for(int i=1;i<=n;i++){
            if(num[i]-j>17)    sum+=(num[i]-j-17)*(num[i]-j-17);
            if(num[i]<j)    sum+=(num[i]-j)*(num[i]-j);    
        }
        ans=min(ans,sum);
    }
    cout<<ans;
}