力扣5389点菜展示表
给你一个数组 orders,表示客户在餐厅中完成的订单,确切地说, orders[i]=[customerNamei,tableNumberi,foodItemi] ,其中 customerNamei 是客户的姓名,tableNumberi 是客户所在餐桌的桌号,而 foodItemi 是客户点的餐品名称。
请你返回该餐厅的 点菜展示表 。在这张表中,表中第一行为标题,其第一列为餐桌桌号 “Table” ,后面每一列都是按字母顺序排列的餐品名称。接下来每一行中的项则表示每张餐桌订购的相应餐品数量,第一列应当填对应的桌号,后面依次填写下单的餐品数量。
注意:客户姓名不是点菜展示表的一部分。此外,表中的数据行应该按餐桌桌号升序排列。
输入: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”
这里讲一下重点,做这种转换形式的题目,首先需要想到的基础方法就是先把他转换成代码易处理的数据结构。比如,这里我就选择了一个KV对表。KEY是桌子号,VALUE又是一个KV表,表示菜名和每样菜的数量。
//表示每张桌子都点了些什么菜
HashMap<Integer,HashMap<String,Integer>> orderBean = new HashMap<>();
先按照输入把这个数据结构创建好,再按照格式输出成题目需要的样子。
题目中菜名需要按照字母顺序排序,订单的顺序需要按照桌子号排序,这里只需要对建表时的菜名表title和桌号表tableNum拍一下序就可以了,并不影响orderBean的操作。
下面上代码:
class Solution {
public List<List<String>> displayTable(List<List<String>> orders) {
List<List<String>> ans = new ArrayList<>();
List<String> title = new ArrayList<>();//用于生成表头子串组
Map<String,Integer> menuPosit = new HashMap<>();//用于记录菜名对应的数组位置
//title.add("Table");
int position = 1;
for(int i=0;i<orders.size();i++) {//生成表头
String dish = orders.get(i).get(2);
if(!title.contains(dish)){//出现了新的菜名
menuPosit.put(dish,position);
position++;
title.add(dish);
}
}
title.sort(new Comparator<String>() {
@Override
public int compare(String o1, String o2) {
return o1.compareTo(o2);
}
});
title.add(0,"Table");
ans.add(title);
//用于按桌号数据化订单信息
HashMap<Integer,HashMap<String,Integer>> orderBean = new HashMap<>();
List<Integer> tableNum = new ArrayList<>();//记录桌号
//按桌号订单在orderBean中填入<桌号,<菜名,菜量>>
for(int i=0;i<orders.size();i++) {
List<String> oneOrder = orders.get(i);
int id = Integer.parseInt(oneOrder.get(1));
if(!tableNum.contains(id))
tableNum.add(id);
String dish = oneOrder.get(2);
if(!orderBean.containsKey(id)) {//还没有这个单号
HashMap<String,Integer> dishNum = new HashMap<>();
dishNum.put(dish,1);
orderBean.put(id,dishNum);
}
else {
HashMap<String, Integer> dishNum = orderBean.get(id);
int count = dishNum.getOrDefault(dish,0);
dishNum.put(dish,count+1);
orderBean.put(id,dishNum);
}
}
tableNum.sort(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o1-o2;
}
});
for(int id : tableNum) {
HashMap<String,Integer> dishNum = orderBean.get(id);//获得id号餐桌的点餐信息
List<String> oneTable = new ArrayList<>(position);
oneTable.add(String.valueOf(id));
for(int i=1;i<title.size();i++) {
String dishCount = String.valueOf(dishNum.getOrDefault(title.get(i),0));
oneTable.add(i,dishCount);
}
ans.add(oneTable);
}
return ans;
}
}