1353. 滑雪场设计
农夫约翰的农场上有 N N N个山峰,每座山的高度都是整数。
在冬天,约翰经常在这些山上举办滑雪训练营。
不幸的是,从明年开始,国家将实行一个关于滑雪场的新税法。
如果滑雪场的最高峰与最低峰的高度差大于 17 17 17,国家就要收税。
为了避免纳税,约翰决定对这些山峰的高度进行修整。
已知,增加或减少一座山峰 x x x单位的高度,需要花费 x 2 x^2 x2的金钱。
约翰只愿意改变整数单位的高度,且每座山峰只能修改一次。
请问,约翰最少需要花费多少钱,才能够使得最高峰与最低峰的高度差不大于 17 17 17。
输入格式
第一行包含整数
N
N
N。
接下来 N N N行,每行包含一个整数,表示一座山的高度。
输出格式
输出一个整数,表示最少花费的金钱。
数据范围
1
≤
N
≤
1000
1≤N≤1000
1≤N≤1000,
数据保证,每座山的初始高度都在
0
∼
100
0∼100
0∼100之间。
输入样例:
5
20
4
1
24
21
输出样例:
18
样例解释
最佳方案为,将高度为 1 的山峰,增加 3 个单位高度,将高度为 24 的山峰,减少 3 个单位高度。
思路:
情况一:
情况二:
由上可以得出结论,修改后的山峰高度一定在0~100
以内。
由此我们可以枚举所有可能的区间[i,i+17]
:
[0,17]
[1,18]
...
[83,100]
- 当前山峰在区间内则不用管。
- 如果小于区间左边界,则
cost += (i - arr[j]) * (i - arr[j])
- 如果大于区间右边界,则
cost += (arr[j] - i - 17) * (arr[j] - i - 17)
Java代码
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int[] arr = new int[n];
for(int i = 0;i < n;i++){
arr[i] = scanner.nextInt();
}
//初始将最小值设置的很大,方便后续对其更新
int minCost = Integer.MAX_VALUE;
for(int i = 0;i + 17 < 100;i++){//枚举每一个可能的区间
int cost = 0;
for(int j = 0;j < n;j++){
if(arr[j] < i) cost += (i - arr[j]) * (i - arr[j]);
if(arr[j] > i + 17) cost += (arr[j] - i - 17) * (arr[j] - i - 17);
}
//更新最小花费
if(cost < minCost) minCost = cost;
}
System.out.println(minCost);
}
}