思路:
情况一:
情况二:
由上可以得出结论,修改后的山峰高度一定在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);
}
}