【牛客网编程题】

牛客网编程题

题目
小红拿到了一个数组 a，每次操作小红可以选择数组中的任意一个数减去 x，小红一共能进行 k 次。小红想在 k 次操作之后，数组的最大值尽可能小。请你返回这个最大值。

原题网址：腾讯音乐娱乐集团2022暑期实习生招聘技术类岗位编程题

思路

最大值操作，可以采用堆排序的思路，在此题中，可以不需要排序，只需要建堆获取当前最大值即可。当然也可以先排序，也可以得到k次操作后的最大值结果

代码

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * @param a int整型ArrayList 
     * @param k int整型 
     * @param x int整型 
     * @return int整型
     */
    public int minMax (ArrayList<Integer> a, int k, int x) {
        // write code here
        // 初始化数据
        ArrayList<Integer> alist = new ArrayList();
        alist.add(0);
        alist.addAll(a);
        
        // 建堆
        for (int i=alist.size()/2; i>0; i--) {
            adjust(alist,i,alist.size());
        }
        
        // k次操作，每次操作完重新建堆
        for (int j=k; j>0; j--) {
            alist.set(1,alist.get(1)-x);
            adjust(alist,1,alist.size());
        }
        return alist.get(1);
        }
    
    public static void adjust(ArrayList<Integer> arrayList, int k, int len) {
        
        arrayList.set(0,arrayList.get(k)); // 用arrayList[0]暂存子树的根节点

        for (int i = 2*k; i < len; i*=2) {  // 根顶元素下坠
            if(i+1 < len && arrayList.get(i) < arrayList.get(i+1)) { //左右子元素比较，取最大的子元素
                i++;
            }
            if( arrayList.get(0) >= arrayList.get(i) ) { // 父元素更大，不做处理
                break;
            } else {
                arrayList.set(k,arrayList.get(i));  // 将子元素替换父元素
                k = i;
            }
        }
        arrayList.set(k,arrayList.get(0));  //将被筛选的结点的值放入最终位置
    }

}

运行结果

时间复杂度分析

根据堆排序算法的时间复杂度分析，建堆时间复杂度为时间复杂度为O(n)，而需要k+1次建堆，故此方法时间复杂度为O(k*n)