1. 程式人生 > 實用技巧 >leetcode-1418 Display Table of Food Orders in a Restaurant

leetcode-1418 Display Table of Food Orders in a Restaurant

Given the array orders, which represents the orders that customers have done in a restaurant. More specifically orders[i][customerNamei,tableNumberi,foodItemi] where customerNamei is the name of the customer, 
tableNumberi is the table customer sit at, and foodItemi is the item customer orders. Return the restaurant's “display table”. The “display table” is a table whose row entries denote how many of each food item each table ordered. The first column is the table number and the remaining columns
correspond to each food item in alphabetical order. The first row should be a header whose first column is “Table”, followed by the names of the food items. Note that the customer names are not part of the
table. Additionally, the rows should be sorted in numerically increasing order.

example:

Input: orders = [["David","3","Ceviche"],["Corina","10","Beef Burrito"],["David","3","Fried Chicken"],["Carla","5","Water"],["Carla","5","Ceviche"],["Rous","3","Ceviche"]]
Output: [["Table","Beef Burrito","Ceviche","Fried Chicken","Water"],["3","0","2","1","0"],["5","0","1","0","1"],["10","1","0","0","0"]] 
Explanation:
The displaying table looks like:
Table,Beef Burrito,Ceviche,Fried Chicken,Water
3    ,0           ,2      ,1            ,0
5    ,0           ,1      ,0            ,1
10   ,1           ,0      ,0            ,0
For the table 3: David orders "Ceviche" and "Fried Chicken", and Rous orders "Ceviche".
For the table 5: Carla orders "Water" and "Ceviche".
For the table 10: Corina orders "Beef Burrito". 

本題思路其實很簡單,就是統計每個桌子各個菜的數量,如果沒有該菜就輸出0。程式碼如下:

 1 class Solution:
 2     def displayTable(self, orders: List[List[str]]) -> List[List[str]]:
 3         tables = {}
 4         names = []
 5         nums = []
 6         for table in orders:
 7             if table[1] in tables:
 8                 if table[2] in
tables[table[1]]: 9 tables[table[1]][table[2]] += 1 10 else: 11 tables[table[1]][table[2]] = 1 12 else: 13 nums.append(int(table[1])) 14 tables[table[1]] = {table[2]:1} 15 if table[2] not in names: 16 names.append(table[2]) 17 names.sort() 18 names.insert(0,"Table") 19 nums.sort() 20 displays = [] 21 displays.append(names) 22 for i in nums: 23 display = [] 24 display.append(str(i)) 25 for j in names: 26 if j in tables[str(i)]: 27 if j != "Table": 28 display.append(str(tables[str(i)][j])) 29 else: 30 if j != "Table": 31 display.append("0") 32 displays.append(display) 33 return displays