牛客--一样的水

50%测试案例未超时,其余超时,但是锻炼思路的一种解法

import java.util.*;


public class Solution {
    /**
     * 
     * @param n int整型 水桶的个数
     * @param q int整型 询问的次数
     * @param a int整型一维数组 n个水桶中初始水的体积
     * @param p int整型一维数组 每次询问的值
     * @return int整型一维数组
     */
    public int[] solve (int n, int q, int[] a, int[] p) {
        // write code here
        Arrays.sort(a);
        int[] ans = new int[q];
        for(int i=0;i<q;i++){
            ans[i] = cost(a,p[i]);
        }
        return ans;
    }
    
    public static int cost(int[] a,int same){
        Map<Integer,Integer> map = new HashMap<>();
        int min = Integer.MAX_VALUE;//记录数组中的最小值
        for(int i=0;i<a.length;i++) {//用map记录每种水量的相同数目
            min = Math.min(min, a[i]);
            int cnt = map.getOrDefault(a[i], 0);
            map.put(a[i], cnt + 1);
            if (cnt + 1 > same) return 0;
        }
        int ans = Integer.MAX_VALUE;
        int minWater = -1;
        for(Integer k : map.keySet()){
            int tmpCost = 0;
            int needChange = same-map.get(k);
            int step = 1;
            boolean findFlag = true;
            while(needChange>0){
                if(k-step < min){//往前面没有水桶了
                    findFlag = false;
                    break;
                }
                if(map.getOrDefault(k-step,0)!=0){//只能往前比较少水的水桶加水
                    int change = Math.min(needChange,map.getOrDefault(k-step,0));
                    needChange -= change;
                    tmpCost += change*step;
                }
                step++;
            }
            if(findFlag && tmpCost < ans){
                ans = tmpCost;//记录最少花费
                minWater = k;//取得相同的水桶数为k
            }
        }
        //按计划改给水桶加水
        int start=0;
        for(;start<a.length;start++){
            if(a[start] == minWater) break;
        }
        int count = same - map.get(minWater);
        while(count>0){
            a[--start] = minWater;
            count--;
        }
        return ans;
    }
}