-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path256-Paint_House.py
51 lines (37 loc) · 1.58 KB
/
256-Paint_House.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
class Solution:
def minCost(self, costs: List[List[int]]) -> int:
memo = {}
def paint( point, color ):
if (point, color) in memo:
return memo[ (point, color) ]
total_cost = costs[point][color]
if point == len(costs) - 1:
pass
elif color == 0:
total_cost += min( paint(point+1, 1), paint(point+1, 2) )
elif color == 1:
total_cost += min( paint(point+1, 0), paint(point+1, 2) )
elif color == 2:
total_cost += min( paint(point+1, 1), paint(point+1, 0) )
memo[ (point, color) ] = total_cost
return total_cost
if len(costs) == 0:
return 0
return min( paint(0,0), paint(0,1), paint(0,2) )
## Time Limit Exceeded
class Solution:
def minCost(self, costs: List[List[int]]) -> int:
def paint( point, color ):
total_cost = costs[point][color]
if point == len(costs) - 1:
pass
elif color == 0:
total_cost += min( paint(point+1, 1), paint(point+1, 2) )
elif color == 1:
total_cost += min( paint(point+1, 0), paint(point+1, 2) )
elif color == 2:
total_cost += min( paint(point+1, 1), paint(point+1, 0) )
return total_cost
if len(costs) == 0:
return 0
return min( paint(0,0), paint(0,1), paint(0,2) )