前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 1801. 积压订单中的订单总数(map)

LeetCode 1801. 积压订单中的订单总数(map)

作者头像
Michael阿明
发布2021-09-06 10:20:29
4630
发布2021-09-06 10:20:29
举报
文章被收录于专栏:Michael阿明学习之路

文章目录

1. 题目

给你一个二维整数数组 orders ,其中每个 orders[i] = [pricei, amounti, orderTypei] 表示有 amounti 笔类型为 orderTypei 、价格为 pricei 的订单。

订单类型 orderTypei 可以分为两种:

  • 0 表示这是一批采购订单 buy
  • 1 表示这是一批销售订单 sell

注意,orders[i] 表示一批共计 amounti 笔的独立订单,这些订单的价格和类型相同。 对于所有有效的 i ,由 orders[i] 表示的所有订单提交时间均早于 orders[i+1] 表示的所有订单。

存在由未执行订单组成的 积压订单 。积压订单最初是空的。提交订单时,会发生以下情况:

  • 如果该订单是一笔采购订单 buy ,则可以查看积压订单中价格 最低 的销售订单 sell 。如果该销售订单 sell 的价格 低于或等于 当前采购订单 buy 的价格,则匹配并执行这两笔订单,并将销售订单 sell 从积压订单中删除。否则,采购订单 buy 将会添加到积压订单中。
  • 反之亦然,如果该订单是一笔销售订单 sell ,则可以查看积压订单中价格 最高 的采购订单 buy 。如果该采购订单 buy 的价格 高于或等于 当前销售订单 sell 的价格,则匹配并执行这两笔订单,并将采购订单 buy 从积压订单中删除。否则,销售订单 sell 将会添加到积压订单中。

输入所有订单后,返回积压订单中的 订单总数 。 由于数字可能很大,所以需要返回对 10^9 + 7 取余的结果。

示例 1:

代码语言:javascript
复制
输入:orders = [[10,5,0],[15,2,1],[25,1,1],[30,4,0]]
输出:6
解释:输入订单后会发生下述情况:
- 提交 5 笔采购订单,价格为 10 。没有销售订单,所以这 5 笔订单添加到积压订单中。
- 提交 2 笔销售订单,价格为 15 。没有采购订单的价格大于或等于 15 ,所以这 2 笔订单添加到积压订单中。
- 提交 1 笔销售订单,价格为 25 。没有采购订单的价格大于或等于 25 ,所以这 1 笔订单添加到积压订单中。
- 提交 4 笔采购订单,价格为 30 。前 2 笔采购订单与价格最低(价格为 15)的 2 笔销售订单匹配,从积压订单中删除这 2 笔销售订单。第 3 笔采购订单与价格最低的 1 笔销售订单匹配,销售订单价格为 25 ,从积压订单中删除这 1 笔销售订单。积压订单中不存在更多销售订单,所以第 4 笔采购订单需要添加到积压订单中。
最终,积压订单中有 5 笔价格为 10 的采购订单,和 1 笔价格为 30 的采购订单。所以积压订单中的订单总数为 6 。

示例 2:

代码语言:javascript
复制
输入:orders = [[7,1000000000,1],[15,3,0],[5,999999995,0],[5,1,1]]
输出:999999984
解释:输入订单后会发生下述情况:
- 提交 109 笔销售订单,价格为 7 。没有采购订单,所以这 109 笔订单添加到积压订单中。
- 提交 3 笔采购订单,价格为 15 。这些采购订单与价格最低(价格为 7 )的 3 笔销售订单匹配,从积压订单中删除这 3 笔销售订单。
- 提交 999999995 笔采购订单,价格为 5 。销售订单的最低价为 7 ,所以这 999999995 笔订单添加到积压订单中。
- 提交 1 笔销售订单,价格为 5 。这笔销售订单与价格最高(价格为 5 )的 1 笔采购订单匹配,从积压订单中删除这 1 笔采购订单。
最终,积压订单中有 (1000000000-3) 笔价格为 7 的销售订单,和 (999999995-1) 笔价格为 5 的采购订单。所以积压订单中的订单总数为 1999999991 ,等于 999999984 % (10^9 + 7) 。
 

提示:
1 <= orders.length <= 10^5
orders[i].length == 3
1 <= pricei, amounti <= 10^9
orderTypei 为 0 或 1

https://leetcode-cn.com/problems/number-of-orders-in-the-backlog/

2. 解题

  • 使用 map ,更新维护,数量为0的时候删除
代码语言:javascript
复制
typedef pair<int,int> pii;
struct cmp{
    bool operator()(int a, int b) const
    {
        return a > b;
    }
};
class Solution {
public:
    int getNumberOfBacklogOrders(vector<vector<int>>& orders) {
        map<int, long long> qsell;
        map<int,long long,cmp> qbuy;
        for(int i = 0; i < orders.size(); ++i)
        {
            int num = orders[i][1];
            int price = orders[i][0];
            if(orders[i][2] == 0)//buy订单
            {
                for(auto it=qsell.begin(); it != qsell.end() && num; )
                {
                    if(price >= it->first)
                    {
                        if(it->second > num)
                        {
                            it->second -= num;
                            num = 0;
                            break;
                        }
                        else
                        {
                            num -= it->second;
                            it->second = 0;
                            qsell.erase(it++);
                        }
                    }
                    else
                        break;
                }
                if(num)
                {
                    qbuy[price] += num;
                }
            }
            else
            {
                for(auto it=qbuy.begin(); it != qbuy.end() && num; )
                {
                    if(price <= it->first)
                    {
                        if(it->second > num)
                        {
                            it->second -= num;
                            num = 0;
                            break;
                        }
                        else
                        {
                            num -= it->second;
                            it->second = 0;
                            qbuy.erase(it++);
                        }
                    }
                    else
                        break;
                }
                if(num)
                {
                    qsell[price] += num;
                }
            }
        }
        long long ans = 0, mod = 1e9+7;
        for(auto it=qsell.begin(); it != qsell.end(); ++it)
        {
            ans = (ans + it->second)%mod;
        }
        for(auto it=qbuy.begin(); it != qbuy.end(); ++it)
        {
            ans = (ans + it->second)%mod;
        }
        return ans;
    }
};

248 ms 62.9 MB C++


我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021/03/21 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 文章目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档