寒假每日一题打卡day17——AcWing 1353. 滑雪场设计

【题目描述】

AcWing 1353. 滑雪场设计

【思路】
山的高度范围为0-100，使用数组下标表示元素大小，其值表示该元素的个数。确定枚举的边界，最小山峰高度i的范围为[min, max - 17 - min]，每一个最小山峰对应着一个高度为j = i+17的最高山峰，需将小于最小山峰高度的山峰修改为i，将高于最高山峰修改为i + 17;

import java.io.*;
class Main{
    static int N = 1010;
    public static void main(String args[]) throws Exception{
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in) );
        int w[] =new int[N];
        int n = Integer.parseInt(bf.readLine());
        int min = Integer.MAX_VALUE, max = Integer.MIN_VALUE;
        for(int i = 0; i < n; i ++)  {
            int a =  Integer.parseInt(bf.readLine());
            w[a]++;
        }
        
        int first = 0;
        for(int i = 0; i <= 100; i++){
            //最小值
           if( w[i] > 0 && first ==0) {
               first ++ ;
               min = i;
           }
           //最大值
           if( w[i] > 0) max = i;
        }
        if( max - min <= 17){
            System.out.println(0);
            System.exit(0);
        }
        int ans = Integer.MAX_VALUE;
        //最小山峰范围：min在 [min, max - 17 - min]
        for(int i = min; i <= max - 17 - min; i ++){
            //最小山峰值为i 修改现有山峰高度小于i的山峰高度为i
            int res = 0;
            for(int j = 0; j < i; j++)
                if( w[j] > 0) res += w[j]*(j - i)*(j - i);
            //最大山峰值为 i + 17
            for(int j = 100; j >= i+17; j--)
                if( w[j] > 0) res += w[j]*(j - i - 17)*(j - i - 17);
            if( res < ans) ans =res;
        }
        System.out.println(ans);
    }
}