Skip to content

Commit

Permalink
Create paint-house-ii.py
Browse files Browse the repository at this point in the history
  • Loading branch information
kamyu104 committed Aug 20, 2015
1 parent 7497acc commit c3db6ce
Showing 1 changed file with 27 additions and 0 deletions.
27 changes: 27 additions & 0 deletions Python/paint-house-ii.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
# Time: O(n * k)
# Space: O(k)

class Solution(object):
def minCostII(self, costs):
"""
:type costs: List[List[int]]
:rtype: int
"""
if not costs:
return 0

n = len(costs)
k = len(costs[0])
min_cost = [costs[0], [0] * k]
for i in xrange(1, n):
min_1st, min_2nd = float("inf"), float("inf")
for j in xrange(k):
if min_1st >= min_cost[(i - 1) % 2][j]:
min_1st, min_2nd = min_cost[(i - 1) % 2][j], min_1st
elif min_2nd >= min_cost[(i - 1) % 2][j]:
min_2nd = min_cost[(i - 1) % 2][j]
for j in xrange(k):
min_j = min_1st if min_cost[(i - 1) % 2][j] != min_1st else min_2nd
min_cost[i % 2][j] = costs[i][j] + min_j

return min(min_cost[(n - 1) % 2])

0 comments on commit c3db6ce

Please sign in to comment.