难度困难156收藏分享切换为英文接收动态反馈
假设 力扣(LeetCode)即将开始 IPO 。为了以更高的价格将股票卖给风险投资公司,力扣 希望在 IPO 之前开展一些项目以增加其资本。 由于资源有限,它只能在 IPO 之前完成最多 k
个不同的项目。帮助 力扣 设计完成最多 k
个不同项目后得到最大总资本的方式。
给你 n
个项目。对于每个项目 i
,它都有一个纯利润 profits[i]
,和启动该项目需要的最小资本 capital[i]
。
最初,你的资本为 w
。当你完成一个项目时,你将获得纯利润,且利润将被添加到你的总资本中。
总而言之,从给定项目中选择 最多 k
个不同项目的列表,以 最大化最终资本 ,并输出最终可获得的最多资本。
答案保证在 32 位有符号整数范围内。
示例 1:
输入:k = 2, w = 0, profits = [1,2,3], capital = [0,1,1] 输出:4 解释: 由于你的初始资本为 0,你仅可以从 0 号项目开始。 在完成后,你将获得 1 的利润,你的总资本将变为 1。 此时你可以选择开始 1 号或 2 号项目。 由于你最多可以选择两个项目,所以你需要完成 2 号项目以获得最大的资本。 因此,输出最后最大化的资本,为 0 + 1 + 3 = 4。
示例 2:
输入:k = 3, w = 0, profits = [1,2,3], capital = [0,1,2] 输出:6
解题思路
思路还是比较简单的,因为总资本数只增不减,当完成一个任务后就把他的利润加到总的资本上面就好了。
利用优先级队列,将符合条件的任务的利润数加入到队列中, 如果本次符合条件的任务利润全部加入到了队列中后,执行出队操作,并把其利润加到总资本中去。
直到任务数达到要求为止。
如果队列为空时,说明已经没有符合条件的任务了(即便是没有达到最大任务量也需要结束程序了。)程序退出。
复习到的点有: 如果两个数组的元素有对应关系时,想要对其中一个数组进行排序,但需要保持另一个数组也同步变化时,可以采用list嵌套数组的形式。
把两个数组的对应元素添加为一个一维数组,然后多个数组添加到list中去。排序的时候利用 Collections.sort方法指定排序方式为原来某数组进行排序。
当完成排序后,遍历该list 同时将profits[i]加入优先队列中,这样能够保证所有的当前某时刻不可满足条件的任务也不被忽略。
代码大佬的代码:
class Solution {
public int findMaximizedCapital(int k, int w, int[] profits, int[] capital) {
List<int[]> list = new ArrayList<>();
for(int i = 0;i<profits.length;i++){
list.add(new int []{capital[i],profits[i]});
}
Collections.sort(list,(a,b)->a[0]-b[0]);
for(int i =0;i<list.size();i++){
System.out.println( list.get(i)[0]+" "+list.get(i)[1]);
}
PriorityQueue<Integer> pq = new PriorityQueue<>((a,b)->(b-a));
int start = 0;
while(k>0){
while( start < list.size() && list.get(start)[0]<=w ){
pq.add(list.get(start)[1]);
start++;
}
if(pq.isEmpty()) break;
w+=pq.poll();
k--;
}
return w;
}}
我一开始的代码:
我写的代码没有用优先队列,也没有对数据进行预处理,就每次从头开始暴力尝试,可谓是很稚嫩的手法。
public int findMaximizedCapital(int k, int w, int[] profits, int[] capital) {
List<int[]> list = new ArrayList<>();
for(int i = 0;i<profits.length;i++){
list.add(new int []{capital[i],profits[i]});
}
Collections.sort(list,(a,b)->a[0]-b[0]);
for(int i =0;i<list.size();i++){
System.out.println( list.get(i)[0]+" "+list.get(i)[1]);
}
while(k>0){
int maxp = 0,maxflag=0;
int start = 0;
while(list.size()>0 && start <list.size() && list.get(start)[0]<=w){
if(list.get(start)[1] >maxp){
maxp = list.get(start)[1];
maxflag = start;
}
start++;
}
w+=maxp;
if(list.size()>0)
list.remove(maxflag);
k--;
start=0;
}
return w;
}
}