雇佣K个工人的最小费用 Minimum Cost to Hire K Workers

2018-10-06 20:17:30

问题描述：

问题求解：

问题规模是10000，已经基本说明是O(nlogn)复杂度的算法，这个复杂度最常见的就是排序算法了，本题确实是使用排序算法来进行进行求解。

本题中指出最后支付的费用和工人的quality相关，也就是说

paid[i] : quality[i] = paid[j] : quality[j] = ratio(固定值)

换言之我们可以按照wage[i] / quality[i]来进行排序，另外还需要维护一个优先队列，大小为K，在超过K之后需要将quality最大的数进行排除。

    public double mincostToHireWorkers(int[] quality, int[] wage, int K) {

        double res = Double.MAX_VALUE;

        int n = quality.length;

        double[][] workers = new double[n][2];

        for (int i = 0; i < n; i++) {

            workers[i][0] = wage[i] * 1.0 / quality[i];

            workers[i][1] = quality[i];

        }

        Arrays.sort(workers, new Comparator<double[]>() {

            @Override

            public int compare(double[] o1, double[] o2) {

                return Double.compare(o1[0], o2[0]);

            }

        });

        PriorityQueue<Double> pq = new PriorityQueue<>();

        double curSum = 0;

        for (double[] worker : workers) {

            curSum += worker[1];

            pq.add(-worker[1]);

            if (pq.size() > K) curSum += pq.poll();

            if (pq.size() == K) res = Math.min(res, curSum * worker[0]);

        }

        return res;

    }