Update Find All Possible Recipes from Given Supplies.java

varunu28 · web-flow · commit 04c96f86a585 · 2022-04-07T19:56:46.000+05:30
diff --git a/Medium/Find All Possible Recipes from Given Supplies.java b/Medium/Find All Possible Recipes from Given Supplies.java
@@ -1,30 +1,29 @@
 class Solution {
-  public List<String> findAllRecipes(String[] recipes, List<List<String>> ingredients,
-      String[] supplies) {
-    Map<String, Set<String>> ingredientToRecipeMap = new HashMap<>();
-    Map<String, Integer> recipeToIngredientCount = new HashMap<>();
+  public List<String> findAllRecipes(String[] recipes, List<List<String>> ingredients, String[] supplies) {
+    Map<String, List<String>> map = new HashMap<>();
+    Map<String, Integer> ingredientCounter = new HashMap<>();
     for (int i = 0; i < recipes.length; i++) {
-      String recipe = recipes[i];
-      List<String> ingredientsForRecipe = ingredients.get(i);
-      recipeToIngredientCount.put(recipe, ingredientsForRecipe.size());
-      for (String ingredient : ingredientsForRecipe) {
-        ingredientToRecipeMap.computeIfAbsent(ingredient, k -> new HashSet<>()).add(recipe);
+      List<String> currentIngredientList = ingredients.get(i);
+      for (String ingredient : currentIngredientList) {
+        map.computeIfAbsent(ingredient, k -> new ArrayList<>()).add(recipes[i]);
+        ingredientCounter.put(recipes[i], ingredientCounter.getOrDefault(recipes[i], 0) + 1);
       }
     }
-    Queue<String> queue = new LinkedList<>(List.of(supplies));
-    Set<String> result = new HashSet<>();
+    Queue<String> queue = new LinkedList<>();
+    Set<String> prepared = new HashSet<>();
+    for (String supply : supplies) {
+      queue.add(supply);
+    }
     while (!queue.isEmpty()) {
-      String ingredient = queue.remove();
-      for (String dependentRecipe : ingredientToRecipeMap.getOrDefault(ingredient,
-          new HashSet<>())) {
-        recipeToIngredientCount.put(dependentRecipe,
-            recipeToIngredientCount.get(dependentRecipe) - 1);
-        if (recipeToIngredientCount.get(dependentRecipe) == 0) {
-          result.add(dependentRecipe);
+      String removed = queue.remove();
+      for (String dependentRecipe : map.getOrDefault(removed, new ArrayList<>())) {
+        ingredientCounter.put(dependentRecipe, ingredientCounter.getOrDefault(dependentRecipe, 0) - 1);
+        if (ingredientCounter.get(dependentRecipe) == 0) {
           queue.add(dependentRecipe);
+          prepared.add(dependentRecipe);
         }
       }
     }
-    return new ArrayList<>(result);
+    return new ArrayList<>(prepared);
   }
 }
