题目
给你一个数组 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"]]
提示:
- 1 <= orders.length <= 5 * 10^4
- orders[i].length == 3
- 1 <= customerNamei.length, foodItemi.length <= 20
- customerNamei 和 foodItemi 由大小写英文字母及空格字符 ’ ’ 组成。
- tableNumberi 是 1 到 500 范围内的整数。
题解
使用哈希表来存储信息
class Solution {
public List<List<String>> displayTable(List<List<String>> orders) {
Map<String, HashMap<String, Integer>> map = new HashMap<>(); //桌号 --- Map<菜名,数量>
Set<String> set = new HashSet<>();
for(int i = 0; i < orders.size(); i++){
HashMap<String, Integer> tableMap = map.getOrDefault(orders.get(i).get(1), new HashMap<String, Integer>()); // 获取桌号对应的Map
int num = tableMap.getOrDefault(orders.get(i).get(2), 0) + 1; // 桌号内的相同菜名数量+1
tableMap.put(orders.get(i).get(2), num); // 将菜名和对应数量放回
map.put(orders.get(i).get(1), tableMap); // 将桌号和对应map放回
set.add(orders.get(i).get(2)); // 添加菜名
}
String[] names = set.toArray(new String[0]);
Arrays.sort(names); // 获取菜名的字典排序
List<List<String>> res = new ArrayList<>();
List<String> list = new ArrayList<>();
list.add("Table");
for(String s : names)
list.add(s);
res.add(list); // 得到第一行List Table + 菜名字典序列
String[] order = map.keySet().toArray(new String[0]); // 获取桌号字符串数组
int[] nums = new int[order.length]; // 将字符串数组转为int数组排序
for(int i = 0; i < order.length; i++)
nums[i] = Integer.valueOf(order[i]);
Arrays.sort(nums);
for(int i = 0; i < nums.length; i++)
order[i] = String.valueOf(nums[i]); // 将排好序的数组放回原桌号字符串数组
// 遍历map,一个map对应一行list
for(int i = 0; i < map.size(); i++){
List<String> newList = new ArrayList<>();
newList.add(order[i]); // 添加桌号
HashMap<String, Integer> tableMap = map.get(order[i]); //从桌号中拿出 菜名和数量 map
for(int j = 1; j < list.size(); j++){ // 遍历所有菜名
newList.add(String.valueOf(tableMap.getOrDefault(list.get(j), 0))); // 存在该菜就返回数量,不存在就返回0
}
res.add(newList); // 添加进结果中
}
return res;
}
}
官解
class Solution {
public List<List<String>> displayTable(List<List<String>> orders) {
Map<Integer, HashMap<String, Integer>> map = new HashMap<>(); //桌号 --- Map<菜名,数量>
Set<String> set = new HashSet<>(); // 存储菜名
for(List<String> order : orders){
set.add(order.get(2)); // 添加菜名
int id = Integer.parseInt(order.get(1));
HashMap<String, Integer> tableMap = map.getOrDefault(id, new HashMap<String, Integer>()); // 获取桌号对应的Map
tableMap.put(order.get(2), tableMap.getOrDefault(order.get(2), 0) + 1); // 将菜名 和 原数量+1放回
map.put(id, tableMap); // 将桌号和对应map放回
}
// 提取餐品名称并按字母顺序排序
int n = set.size();
List<String> names = new ArrayList<String>();
for(String name : set){
names.add(name);
}
Collections.sort(names); // 获取菜名的字典排序
// 提取桌号,并按桌号升序排列
int m = map.size();
List<Integer> ids = new ArrayList<Integer>();
for(int id : map.keySet()){
ids.add(id);
}
Collections.sort(ids);
// 填写点菜展示表
List<List<String>> table = new ArrayList<List<String>>();
List<String> header = new ArrayList<>();
header.add("Table");
for(String name : names){
header.add(name);
}
table.add(header);
// 遍历map,一个map对应一行list
for(int i = 0; i < m; i++){
int id = ids.get(i);
List<String> row = new ArrayList<>();
HashMap<String, Integer> cnt = map.get(id); //从桌号中拿出 菜名和数量 map
row.add(Integer.toString(id));
for(int j = 0; j < n; j++){ // 遍历所有菜名
row.add(String.valueOf(cnt.getOrDefault(names.get(j), 0))); // 存在该菜就返回数量,不存在就返回0
}
table.add(row); // 添加进结果中
}
return table;
}
}