leetcode 502. IPO

最新推荐文章于 2025-05-09 08:18:45 发布

原创最新推荐文章于 2025-05-09 08:18:45 发布 · 716 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 2 个专栏收录

62 篇文章

订阅专栏

数据结构

19 篇文章

订阅专栏

本文探讨了multiset与priority_queue在实现大小顶堆的区别，通过两个代码示例对比了两者在删除元素时的行为差异，并提供了具体的C++实现案例。

multiset 与 priority_queue

联系：可以实现大小顶堆；

区别：删除元素时，multiset 会把相同元素值都删除！！

代码 1 ：（Accept）

class Solution {
public:
    int findMaximizedCapital(int k, int W, vector<int>& Profits, vector<int>& Capitals) {
        multimap<int,int> mp;
        for (int i = 0; i < Capitals.size(); ++i)
            mp.insert (make_pair(Capitals[i], Profits[i]));
        priority_queue<int> heap;
        while (k--){
            for (auto iter = mp.begin(); iter != mp.end();){
                if (iter->first <= W){
                    heap.push(iter->second); 
                    iter = mp.erase (iter);
                }
                else 
                    break;
            }
            if (!heap.empty()){
                W += heap.top();
                heap.pop();
            }
            else 
                break;
        }
        return W;
    }
};

代码 2 ：（Wrong Answer）

class Solution {
public:
    int findMaximizedCapital(int k, int W, vector<int>& Profits, vector<int>& Capitals) {
        multimap<int,int> mp;
        for (int i = 0; i < Capitals.size(); ++i)
            mp.insert (make_pair(Capitals[i], Profits[i]));
        multiset<int> st;
        while (k--){
            for (auto iter = mp.begin(); iter != mp.end();){
                if (iter->first <= W){
                    st.insert (iter->second); 
                    iter = mp.erase (iter);
                }
                else 
                    break;
            }
            if (!st.empty()){
                W += *st.rbegin();
                st.erase (*st.rbegin()); //   st.erase (--st.end()); 这样写就对了
            }
            else 
                break;
        }
        return W;
    }
};