力扣502-IPO-hard-贪心,优先队列 -java

力扣502-IPO-hard-贪心,优先队列 -java_第1张图片
思路:
本题可以先构造处一个模型,我们的目的在于满足小于等于w的基础上,使得加上profits最大即可,因为是纯利润,不需要考虑w-capital的情况,所以我们只需要构造一个大根堆将所有满足条件的profits加入到堆中即可。st. w>= capitals[i].

代码:

// @author:hairu,WU, fdu
class Solution {
    public int findMaximizedCapital(int k, int w, int[] profits, int[] capital) {
        int n = profits.length;
        int[][] projects = new int[n][2];

        for(int i=0;i<n;i++){
            projects[i][0] = capital[i];
            projects[i][1] = profits[i];
        }

        // 按照启动资金从小到大排序
        Arrays.sort(projects, (p1, p2)->{return p1[0]-p2[0];});
        // 建立大根堆
        PriorityQueue<Integer> queue = new PriorityQueue<Integer>((x,y)->{return y-x;});

        // 开始遍历 
        int index = 0;
        for(int i=0;i<k;i++){
            while(index < n && projects[index][0]<=w){
                queue.add(projects[index][1]);
                index++;
            }

            // 取出元素
            if(!queue.isEmpty()){
                w+=queue.poll();
            }else{  
                break;
            }

        }

        return w;
    }
}

力扣502-IPO-hard-贪心,优先队列 -java_第2张图片

你可能感兴趣的:(算法,java,数据结构)