LeetCode 1801积压订单中的订单总数堆 HERODING的LeetCode之路

Posted 2023-02-27 HERODING23

解题思路：
这道题很有意思，所以想拿来说说。这是一道读题题，比较符合现实中的生活场景，首先根据题目的含义，等于间接提醒coder要构建两个堆，一个最大堆，一个最小堆，最大堆存放积压的采购订单，最小堆存放积压的销售订单。从真实场景中考虑，对于一个采购订单，如果能以更低价格购入商品是最划算的，所以每次考虑的都是销售订单堆中最少价格的订单，当然如果销售订单中最小价格也比采购订单贵，那么就不匹配了，剩下的采购订单只能积压到采购订单堆中。对于一个销售订单，如果能以更高的价格被采购是最好的，因此每次考虑的都是采购订单堆中最高价格的订单。
那么整体的解题思路也是如此，定义一个最大堆和一个最小堆，定义变量amount存储积压的订单数。变量orders数组，首先判断是销售订单还是采购订单，以销售订单为例，如果当前采购订单堆非空且top小于销售订单价格，那么就不断抵消订单，直到当前销售订单空或者采购订单没有能够匹配的，剩下的销售订单要放回积压销售订单堆中，这个过程中的关键在于更新积压订单的数量，详细可以看代码。采购订单也是同样的道理，代码如下：

class Solution 
private:
    const int MOD = 1000000007;
public:
    int getNumberOfBacklogOrders(vector<vector<int>>& orders) 
        int amount = 0;
        priority_queue<pair<int, int>> buy;
        priority_queue<pair<int, int>, vector<pair<int, int>>, greater<>> sell;
        for(auto& order : orders) 
            // 如果是采购订单
            if(order[2] == 0) 
                while(order[1] && !sell.empty() && sell.top().first <= order[0]) 
                    auto [price, num] = sell.top();
                    sell.pop();
                    if(order[1] >= num) 
                        order[1] -= num;
                        amount = (amount - num + MOD) % MOD;
                     else 
                        num -= order[1];
                        amount = (amount - order[1] + MOD) % MOD;
                        order[1] = 0;
                        sell.push(price, num);
                    
                
                if(order[1]) 
                    buy.push(order[0], order[1]);
                    amount = (amount + order[1]) % MOD;
                
             else // 如果是出售订单
                while(order[1] && !buy.empty() && buy.top().first >= order[0]) 
                    auto [price, num] = buy.top();
                    buy.pop();
                    if(order[1] >= num) 
                        order[1] -= num;
                        amount = (amount - num + MOD) % MOD;
                     else 
                        num -= order[1];
                        amount = (amount - order[1] + MOD) % MOD;
                        order[1] = 0; 
                        buy.push(price, num);
                    
                
                if(order[1]) 
                    sell.push(order[0], order[1]);
                    amount = (amount + order[1]) % MOD;
                
            
        
        return amount;
    
;