【LeetCode/力扣】1418. 点菜展示表

1 题目描述

题目链接：https://leetcode-cn.com/problems/display-table-of-food-orders-in-a-restaurant/

---

给你一个数组 orders，表示客户在餐厅中完成的订单，确切地说， orders[i]=[customerNamei,tableNumberi,foodItemi] ，其中 customerNamei 是客户的姓名，tableNumberi 是客户所在餐桌的桌号，而 foodItemi 是客户点的餐品名称。

请你返回该餐厅的 点菜展示表 。在这张表中，表中第一行为标题，其第一列为餐桌桌号 “Table” ，后面每一列都是按字母顺序排列的餐品名称。接下来每一行中的项则表示每张餐桌订购的相应餐品数量，第一列应当填对应的桌号，后面依次填写下单的餐品数量。

注意：客户姓名不是点菜展示表的一部分。此外，表中的数据行应该按餐桌桌号升序排列。

---

示例 1：

输入：orders = [[“David”,“3”,“Ceviche”],[“Corina”,“10”,“Beef Burrito”],[“David”,“3”,“Fried Chicken”],[“Carla”,“5”,“Water”],[“Carla”,“5”,“Ceviche”],[“Rous”,“3”,“Ceviche”]]
输出：[[“Table”,“Beef Burrito”,“Ceviche”,“Fried Chicken”,“Water”],[“3”,“0”,“2”,“1”,“0”],[“5”,“0”,“1”,“0”,“1”],[“10”,“1”,“0”,“0”,“0”]]
解释：
点菜展示表如下所示：
Table,Beef Burrito,Ceviche,Fried Chicken,Water
3 ,0 ,2 ,1 ,0
5 ,0 ,1 ,0 ,1
10 ,1 ,0 ,0 ,0
对于餐桌 3：David 点了 “Ceviche” 和 “Fried Chicken”，而 Rous 点了 “Ceviche”
而餐桌 5：Carla 点了 “Water” 和 “Ceviche”
餐桌 10：Corina 点了 “Beef Burrito”
示例 2：

输入：orders = [[“James”,“12”,“Fried Chicken”],[“Ratesh”,“12”,“Fried Chicken”],[“Amadeus”,“12”,“Fried Chicken”],[“Adam”,“1”,“Canadian Waffles”],[“Brianna”,“1”,“Canadian Waffles”]]
输出：[[“Table”,“Canadian Waffles”,“Fried Chicken”],[“1”,“2”,“0”],[“12”,“0”,“3”]]
解释：
对于餐桌 1：Adam 和 Brianna 都点了 “Canadian Waffles”
而餐桌 12：James, Ratesh 和 Amadeus 都点了 “Fried Chicken”
示例 3：

输入：orders = [[“Laura”,“2”,“Bean Burrito”],[“Jhon”,“2”,“Beef Burrito”],[“Melissa”,“2”,“Soda”]]
输出：[[“Table”,“Bean Burrito”,“Beef Burrito”,“Soda”],[“2”,“1”,“1”,“1”]]

2 代码/C++

• 方法1：采用两个vector储存桌号和菜名，然后排序，再分别建立字典，然后按照坐标法遍历orders得到结果，需要注意的是桌号排序需要转化为int

class Solution {
public:
    vector<vector<string>> displayTable(vector<vector<string>>& orders) {
    // 1. 采用两个vector储存桌号和菜名
        vector<int>tables;
        vector<string>foods;
        for (auto &p : orders){
            if (find(tables.begin(),tables.end(),stoi(p[1])) == tables.end()){
                tables.push_back(stoi(p[1]));
            }            
            if (find(foods.begin(),foods.end(),p[2]) == foods.end()){
                foods.push_back(p[2]);
            }
        }     
   // 2. 排序
        sort(tables.begin(),tables.end(),less<int>());
        sort(foods.begin(),foods.end(),less<string>());
        vector<vector<string> > res(tables.size()+1, vector<string>(foods.size()+1,"0"));
  // 3. 分别建立字典
        unordered_map<string,int>fmap;
        unordered_map<string,int>tmap;
        res[0][0] = "Table";
        for (int i=1; i<res[0].size(); ++i){
            res[0][i] = foods[i-1];
            fmap[foods[i-1]] = i;
        }        
        for (int i=1; i<tables.size()+1; ++i){
            res[i][0] = to_string(tables[i-1]);
            tmap[to_string(tables[i-1])] = i;
        }
   // 4. 坐标法遍历orders得到结果
        for (auto &p : orders){
            int r = tmap[p[1]], c = fmap[p[2]];
            res[r][c] = to_string(stoi(res[r][c])+1);
        }
        return res;
    }
};

• 方法2：利用set内部自动按照红黑树排序的特点（map也是），省去排序过程，优化程序速度

class Solution {
public:
    vector<vector<string>> displayTable(vector<vector<string>>& orders) {
        set<int>tables;
        set<string>foods;
        for (auto &p : orders){
            tables.insert(stoi(p[1]));    
            foods.insert(p[2]);
        }        
        vector<vector<string> > res(tables.size()+1, vector<string>(foods.size()+1,"0"));
        unordered_map<string,int>fmap;
        unordered_map<string,int>tmap;
        res[0][0] = "Table";
        int i = 1;
        for (auto &p : foods){
            res[0][i] = p;
            fmap[p] = i;
            ++i;
        }       
        i = 1; 
        for (auto &p : tables){
            res[i][0] = to_string(p);
            tmap[to_string(p)] = i;
            ++i;
        }
        
        for (auto &p : orders){
            int r = tmap[p[1]], c = fmap[p[2]];
            res[r][c] = to_string(stoi(res[r][c])+1);
        }
        return res;
    }
};

3 代码解析

• 这个题目不是很难，目的是考察使用C++来进行数据的处理和汇总
• 需要灵活使用vector、set、map等数据容器，这里使用set可以实现自动排序