Update minimum-time-to-finish-the-race.py

Author: kamyu104

Python/minimum-time-to-finish-the-race.py

@@ -24,6 +24,5 @@ def ceil_log2(x):
                 cnt += 1
         dp2 = [float("inf")]*numLaps  # dp2[i]: min time to complete i+1 laps with changing zero or more tires
         for i in xrange(numLaps):
-            for j in xrange(min(i+1, len(dp))):
-                dp2[i] = min(dp2[i], (dp2[i-j-1]+changeTime if i-j-1 >= 0 else 0)+dp[j])
+            dp2[i] = min((dp2[i-j-1]+changeTime if i-j-1 >= 0 else 0)+dp[j] for j in xrange(min(i+1, len(dp))))
         return dp2[-1]