解题思路:
这道题很有意思,所以想拿来说说。这是一道读题题,比较符合现实中的生活场景,首先根据题目的含义,等于间接提醒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;
}
};