image_2022-05-23_17-57-27.png
87.1 KB
#medium
#N1418. Display Table of Food Orders in a Restaurant
problem link
#solution
#N1418. Display Table of Food Orders in a Restaurant
problem link
#solution
class Solution {
public static List<List<String>> displayTable(List<List<String>> orders) {
int table; String food;
Map<Integer, TreeMap<String, Integer>> map = new TreeMap<>();
Set<String> set = new TreeSet<>();
for (List<String> order : orders) {
table = Integer.parseInt(order.get(1));food = order.get(2);
set.add(food);
TreeMap<String, Integer> foodMap;
if (map.get(table) == null) foodMap = new TreeMap<>();
else foodMap = map.get(table);
foodMap.put(food, foodMap.getOrDefault(food, 0) +1);
map.put(table, foodMap);
}