面试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
相关推荐
零雲3 分钟前
java面试:可以讲一讲sychronized和ReentrantLock的异同点吗
java·开发语言·面试
crystal_pin23 分钟前
前端流式解析chunk数据思路
面试
甜瓜看代码31 分钟前
面试---h5秒开优化
面试
微笑尅乐34 分钟前
神奇的位运算——力扣136.只出现一次的数字
java·算法·leetcode·职场和发展
吃着火锅x唱着歌1 小时前
LeetCode 3105.最长的严格递增或递减子数组
算法·leetcode·职场和发展
测试19981 小时前
Web自动化测试之测试用例流程设计
自动化测试·软件测试·python·selenium·测试工具·职场和发展·测试用例
zcychong1 小时前
如何让A、B、C三个线程按严格顺序执行(附十一种解)?
java·面试
吃着火锅x唱着歌2 小时前
LeetCode 2765.最长交替子数组
算法·leetcode·职场和发展
墨染点香2 小时前
LeetCode 刷题【91. 解码方法】
算法·leetcode·职场和发展
卡卡酷卡BUG4 小时前
Redis 面试常考问题(高频核心版)
java·redis·面试