add if condition to find if we're at the best scenario leading to lowest sandess level

Author: HassanOHOsman

implement-laptop-allocation/main.py

@@ -36,7 +36,7 @@ def sadness(person: Person, laptop: Laptop) ->int:
 def allocate_laptops(people: List[Person], laptops: List[Laptop]) -> Dict[Person, Laptop]:
 
     best_scenario_assignment = None
-    best_scenario_sadness = float("inf")
+    best_total_sadness = float("inf")
 
     for scenario in scenarios(laptops):
         total_sadness = 0
@@ -45,3 +45,9 @@ def allocate_laptops(people: List[Person], laptops: List[Laptop]) -> Dict[Person
         for person, laptop in zip(people, scenario):
             total_sadness += sadness(person, laptop)
 
+        #Check if this is the best scenario so far
+        if total_sadness < best_total_sadness:
+            best_total_sadness = total_sadness
+            best_scenario_assignment = scenario
+
+