502. IPO

LeetCode

• 题目地址：https://leetcode.com/problems/ipo/#/description
• 问题描述：大概意思就是给一组项目，分别需要启动资金C[i]和收益P[i]，问给定初始资金w和最多k次项目，最多能收益多少
• 解题思路：贪心算法，每次找到当前有资本投资的又收益最大的，这样的过程执行k次，也是很自然的想到先用qsort排序，再k次循环找到当前收益最大的且没有投资过的项目，但是这样想法超时了，于是参考答案，答案是维护一个优先队列，相当于一个比W小的项目收益的最大堆，一共有n个项目，每个项目最多被插入删除1次，复杂度为O(nlogn)，另外一个multiset或者multimap存储比当前资本大的项目，每个项目最多被插入删除1次，也是O(nlogn)，k较大的时候比之前的思路要快
• 代码：
  这个在k=n=50000的时候超时，时间复杂度是O(nlogn+kn)，k接近n的时候是O(n^2)

class Solution {
    public:
        int findMaximizedCapital(int k, int W, vector<int>& Profits, vector<int>& Capital) {
            qsort(Profits,Capital,0,Profits.size()-1);
            //print after sort
            /*
            for (int i = 0; i < Profits.size(); i++) {
                cout << "p & c : " << Profits[i] << "," << Capital[i] << endl;
            }*/
            //k times
            int size = Profits.size();
            bool b[size] = {false};  //visit i if b[i] is true
            for (int i = 0; i < k; i++) {
               for (int j = 0; j < size; j++) {
                   if (!b[j] && Capital[j] <= W) {
                       W += Profits[j];
                       b[j] = true;
                       break;
                   }
               } 
            }
            return W;
        }
        void qsort(vector<int>& p, vector<int>& c, int begin, int end) {
            if (begin > end) return;
            int pivot = p[begin];
            int i = begin + 1;
            int j = end;
            //从大到小
            while (i <= j) {
                while (p[i] > pivot && i <= j)
                    i++;
                while (p[j] <= pivot && i <= j)
                    j--;
                if (i <= j) {
                    swap(p[i],p[j]);
                    swap(c[i],c[j]);
                }
            }
            swap(p[begin],p[j]);
            swap(c[begin],c[j]);
            qsort(p,c,begin,j-1);
            qsort(p,c,j+1,end);
        }
};

思路2代码如下：

class Solution {
public:
    int findMaximizedCapital(int k, int W, vector<int>& Profits, vector<int>& Capital) {
        priority_queue<int> q;
        multimap<int,int> ms;
        //construct q
        for (int i = 0, size = Profits.size(); i < size; i++) {
            if (Capital[i] <= W)
                q.push(Profits[i]);
            else
                ms.emplace(Capital[i],Profits[i]);
        }
        //k times operation
        for (int i = 0; i < k; i++) {
            if (!q.empty()) {
                int max = q.top();
                q.pop();
                W += max;
                for (multimap<int,int>::iterator it = ms.begin(), end = ms.end(); it != end; it++) {
                    if (it->first <= W) {
                        q.push(it->second);
                        ms.erase(it);
                    }
                }
            }
            else
                break;
        }
        return W;
    }
};