Update minimum-total-distance-traveled.py

rhshriva · Nov 6, 2022 · 481294d · 481294d
1 parent e15fe6e
commit 481294d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/Python/minimum-total-distance-traveled.py b/Python/minimum-total-distance-traveled.py
@@ -17,14 +17,12 @@ def minimumTotalDistance(self, robot, factory):
         for i in xrange(len(factory)):
             prefix = 0
             dq = collections.deque([(0, -1)])
-            new_dp = [float("inf")]*(len(robot))
             for j in xrange(len(robot)):
                 prefix += abs(robot[j]-factory[i][0])
                 if j-dq[-1][1] == factory[i][1]+1:
                     dq.pop()
                 while dq and dq[0][0] >= dp[j]-prefix:
                     dq.popleft()
                 dq.appendleft((dp[j]-prefix, j))
-                new_dp[j] = dq[-1][0]+prefix
-            dp = new_dp
+                dp[j] = dq[-1][0]+prefix
         return dp[-1]