洛谷 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。

【分析】

依旧是枚举，不多说了。

【代码】

 1 #include <bits/stdc++.h>
 2 #define inf 0x7fffffff
 3 using namespace std;
 4
 5 int n, h[1005], ans=inf, tot;
 6
 7 inline int read() {
 8     int x=0, w=1;
 9     char ch=0;
10     while (ch<‘0‘ || ch>‘9‘) {
11         if (ch==‘-‘)
12             w=-1;
13         ch=getchar();
14     }
15     while (ch>=‘0‘ && ch<=‘9‘)
16         x=(x<<3)+(x<<1)+ch-‘0‘, ch=getchar();
17     return x*w;
18 }
19
20 int main() {
21     cin >> n;
22     for (int i=1;i<=n;++i) scanf("%d", &h[i]);
23     for (int i=0;i<=100;++i) {
24         tot=0;
25         for (int j=1;j<=n;++j)
26             if ((h[j]<i && i-h[j]>17))
27                 tot+=(i-h[j]-17)*(i-h[j]-17);
28             else if (h[j]>i)
29                 tot+=(h[j]-i)*(h[j]-i);
30         ans=min(tot, ans);
31     }
32     cout << ans << endl;
33 }