diff --git a/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/README.md b/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/README.md new file mode 100644 index 000000000..b43a972d1 --- /dev/null +++ b/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/README.md @@ -0,0 +1,46 @@ +# [1418.Display Table of Food Orders in a Restaurant][title] + +## Description +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"]] +``` + +## 结语 + +如果你同我一样热爱数据结构、算法、LeetCode，可以关注我 GitHub 上的 LeetCode 题解：[awesome-golang-algorithm][me] + +[title]: https://leetcode.com/problems/display-table-of-food-orders-in-a-restaurant +[me]: https://github.com/kylesliu/awesome-golang-algorithm diff --git a/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/Solution.go b/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/Solution.go index d115ccf5e..42b4ccd01 100755 --- a/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/Solution.go +++ b/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/Solution.go @@ -1,5 +1,46 @@ package Solution -func Solution(x bool) bool { - return x +import ( + "sort" + "strconv" +) + +func Solution(orders [][]string) [][]string { + foodHeaders := make([]string, 0) + foodHeaderIndex := make(map[string]struct{}) + tables := make([]string, 0) + tableFood := make(map[string]map[string]int) + + for _, order := range orders { + t, f := order[1], order[2] + if _, ok := foodHeaderIndex[f]; !ok { + foodHeaders = append(foodHeaders, f) + foodHeaderIndex[f] = struct{}{} + } + + if _, ok := tableFood[t]; !ok { + tables = append(tables, t) + tableFood[t] = make(map[string]int) + } + tableFood[t][f]++ + } + sort.Strings(foodHeaders) + foodHeaders = append([]string{"Table"}, foodHeaders...) + sort.Slice(tables, func(i, j int) bool { + a, _ := strconv.Atoi(tables[i]) + b, _ := strconv.Atoi(tables[j]) + return a < b + }) + + ans := make([][]string, 0) + ans = append(ans, foodHeaders) + for _, t := range tables { + line := make([]string, len(foodHeaders)) + line[0] = t + for i := 1; i < len(line); i++ { + line[i] = strconv.Itoa(tableFood[t][foodHeaders[i]]) + } + ans = append(ans, line) + } + return ans } diff --git a/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/Solution_test.go b/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/Solution_test.go index 14ff50eb4..26d8024ec 100755 --- a/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/Solution_test.go +++ b/leetcode/1401-1500/1418.Display-Table-of-Food-Orders-in-a-Restaurant/Solution_test.go @@ -10,12 +10,41 @@ func TestSolution(t *testing.T) { // 测试用例 cases := []struct { name string - inputs bool - expect bool + inputs [][]string + expect [][]string }{ - {"TestCase", true, true}, - {"TestCase", true, true}, - {"TestCase", false, false}, + {"TestCase1", [][]string{ + {"David", "3", "Ceviche"}, + {"Corina", "10", "Beef Burrito"}, + {"David", "3", "Fried Chicken"}, + {"Carla", "5", "Water"}, + {"Carla", "5", "Ceviche"}, + {"Rous", "3", "Ceviche"}, + }, [][]string{ + {"Table", "Beef Burrito", "Ceviche", "Fried Chicken", "Water"}, + {"3", "0", "2", "1", "0"}, + {"5", "0", "1", "0", "1"}, + {"10", "1", "0", "0", "0"}, + }}, + {"TestCase2", [][]string{ + {"James", "12", "Fried Chicken"}, + {"Ratesh", "12", "Fried Chicken"}, + {"Amadeus", "12", "Fried Chicken"}, + {"Adam", "1", "Canadian Waffles"}, + {"Brianna", "1", "Canadian Waffles"}, + }, [][]string{ + {"Table", "Canadian Waffles", "Fried Chicken"}, + {"1", "2", "0"}, + {"12", "0", "3"}, + }}, + {"TestCase3", [][]string{ + {"Laura", "2", "Bean Burrito"}, + {"Jhon", "2", "Beef Burrito"}, + {"Melissa", "2", "Soda"}, + }, [][]string{ + {"Table", "Bean Burrito", "Beef Burrito", "Soda"}, + {"2", "1", "1", "1"}, + }}, } // 开始测试 @@ -30,10 +59,10 @@ func TestSolution(t *testing.T) { } } -// 压力测试 +// 压力测试 func BenchmarkSolution(b *testing.B) { } -// 使用案列 +// 使用案列 func ExampleSolution() { }