【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 specificallyorders[i]=[customerNamei,tableNumberi,foodItemi]wherecustomerNameiis the name of the customer,tableNumberiis the table customer sit at, andfoodItemiis 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 1:
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".Example 2:
Input: orders = [["James","12","Fried Chicken"],["Ratesh","12","Fried Chicken"],["Amadeus","12","Fried Chicken"],
["Adam","1","Canadian Waffles"],["Brianna","1","Canadian Waffles"]] Output: [["Table","Canadian Waffles","Fried Chicken"],["1","2","0"],["12","0","3"]] Explanation: For the table 1: Adam and Brianna order "Canadian Waffles". For the table 12: James, Ratesh and Amadeus order "Fried Chicken".Example 3:
Input: orders = [["Laura","2","Bean Burrito"],["Jhon","2","Beef Burrito"],["Melissa","2","Soda"]] Output: [["Table","Bean Burrito","Beef Burrito","Soda"],["2","1","1","1"]]Constraints:
1 <= orders.length <= 5 * 10^4orders[i].length == 31 <= customerNamei.length, foodItemi.length <= 20customerNameiandfoodItemiconsist of lowercase and uppercase English letters and the space character.tableNumberiis a valid integer between1and500.
解题思路:这种题目也没什么好说的,排序就好了。
代码如下:
class Solution(object): def displayTable(self, orders): """ :type orders: List[List[str]] :rtype: List[List[str]] """ dic = {} tables = [] foods = [] for customerName,tableNumber,foodItem in orders: tableNumber = int(tableNumber) if tableNumber not in tables:tables.append(tableNumber) if foodItem not in foods:foods.append(foodItem) dic[(tableNumber,foodItem)] = dic.setdefault((tableNumber,foodItem),0) + 1 tables.sort() foods.sort() headers = ['Table'] + foods res = [headers] for t in tables: table = [str(t)] for f in foods: if (t,f) not in dic: table.append('0') else: table.append(str(dic[t,f])) res.append(table) return res
浙公网安备 33010602011771号