Time: 53 ms (63.71%), Space: 14 MB (17.48%) - LeetHub

r-rahul1 · Mar 25, 2022 · 9a75cd4 · 9a75cd4
1 parent 25df5b0
commit 9a75cd4
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/1029-two-city-scheduling/1029-two-city-scheduling.py b/1029-two-city-scheduling/1029-two-city-scheduling.py
@@ -0,0 +1,15 @@
+class Solution:
+    def twoCitySchedCost(self, costs: List[List[int]]) -> int:
+        diff = []
+        for a,b in costs:
+            diff.append([b-a,a,b])
+
+        diff.sort()
+        res = 0
+        for i in range(len(diff)):
+            if i < len(diff)//2:
+                res += diff[i][2]
+            else:
+                res += diff[i][1]
+
+        return res