display_table_of_food_orders_in_a_restaruant.py

# 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 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"]]

class Solution:
    def displayTable(self, orders: List[List[str]]) -> List[List[str]]:
        food_items = []
        table_nums = []
        map_orders = {}
        
        for order in orders:
            if order[2] not in food_items:
                food_items.append(order[2])
            if int(order[1]) not in table_nums:
                table_nums.append(int(order[1]))
            new_order = (int(order[1]), order[2])
            if new_order not in map_orders:
                map_orders[new_order] = 1
            else:
                map_orders[new_order] += 1
        
        food_items.sort()
        table_nums.sort()
        food_items.insert(0, 'Table')
        
        answer = []
        answer.append(food_items)
        
        for i in range(len(table_nums)):
            new_entry = []
            for j in range(len(food_items)):
                if j == 0:
                    new_entry.append(str(table_nums[i]))
                else:
                    entry_tuple = (table_nums[i], food_items[j])
                    if entry_tuple in map_orders:
                        new_entry.append(str(map_orders[entry_tuple]))
                    else:
                        new_entry.append('0')
                        
            answer.append(new_entry)
            new_entry = []
        
        return answer