leetcode 502. IPO

最新推荐文章于 2021-09-08 15:15:19 发布

shiyang6017

最新推荐文章于 2021-09-08 15:15:19 发布

阅读量648

点赞数 1

分类专栏：算法数据结构

本文链接：https://blog.csdn.net/shiyang6017/article/details/54952709

版权

算法同时被 2 个专栏收录

62 篇文章 0 订阅

订阅专栏

数据结构

19 篇文章 0 订阅

订阅专栏

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;
    }
};