面试150 IPO

思路

首先,将每个项目的启动资本需求和对应的利润配对,组成一个二元组列表,并根据所需资本从小到大进行排序。这样可以确保在遍历项目列表时,能按所需资本的升序处理。接着,使用一个最大堆(通过在堆中存入利润的负值来实现)来维护当前资本下所有可选项目的利润。在每一轮(最多进行 k 轮)中,程序会将当前可承受的所有项目(即资本需求不超过当前拥有资本的项目)加入最大堆,然后从中选择利润最高的项目(堆顶元素),执行该项目并将其利润加到当前资本上。如果在某一轮没有任何可执行的项目,算法会提前退出。最终,返回选择最多 k 个项目后所能获得的最大资本。该算法通过贪心策略结合堆结构,兼顾了效率和最优选择。

python 复制代码
import heapq
from typing import List

class Solution:
    def findMaximizedCapital(self, k: int, w: int, profits: List[int], capital: List[int]) -> int:
        num_projects = len(profits)
        project_list = []
        
        # 将每个项目的资本需求和对应利润配对
        for i in range(num_projects):
            project_list.append([capital[i], profits[i]])
        
        # 按所需资本升序排序
        project_list.sort()
        
        max_profit_heap = []  # 最大堆,用于存储可选项目的利润(取负值实现最大堆)
        project_index = 0
        available_capital = w
        
        for _ in range(k):
            # 将当前资本可承受的项目加入最大堆
            while project_index < num_projects and project_list[project_index][0] <= available_capital:
                heapq.heappush(max_profit_heap, -project_list[project_index][1])
                project_index += 1
            
            if not max_profit_heap:
                break  # 没有可选项目,退出循环
            
            # 选择利润最高的项目
            max_profit = -heapq.heappop(max_profit_heap)
            available_capital += max_profit
        
        return available_capital
相关推荐
我是渣哥12 小时前
Java String vs StringBuilder vs StringBuffer:一个性能优化的探险故事
java·开发语言·jvm·后端·算法·职场和发展·性能优化
睡不醒的kun18 小时前
leetcode算法刷题的第二十六天
数据结构·c++·算法·leetcode·职场和发展·贪心算法
DarkLONGLOVE20 小时前
JS魔法中介:Proxy和Reflect为何形影不离?
前端·javascript·面试
jingling55520 小时前
【高级】系统架构师 | 信息系统战略规划、EAI 与新技术
程序人生·职场和发展·系统架构·学习方法·程序员创富
yinke小琪20 小时前
分库分表后,主键 ID 如何优雅生成?
java·后端·面试
Q741_1471 天前
C++ 面试高频考点 力扣 153. 寻找旋转排序数组中的最小值 二分查找 题解 每日一题
c++·算法·leetcode·面试·二分查找
小高0071 天前
🚄 前端人必收:5 分钟掌握 ES2025 超实用语法
前端·javascript·面试
麦格芬2301 天前
LeetCode 994 腐烂的橘子
算法·leetcode·职场和发展
Java中文社群1 天前
面试官:如何提升项目并发性能?
java·后端·面试
前端缘梦1 天前
深入浅出 Vue 的 Diff 算法:最小化 DOM 操作的魔法
前端·vue.js·面试