Skip to content

Commit ede276e

Browse files
authored
Create house-robber-II.py
Same as previous house robber solution but houses are arranged circularly
1 parent 0e00f47 commit ede276e

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

house-robber-II.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
class Solution(object):
2+
def rob(self, nums):
3+
return max(nums[0] + self.circle(nums[2:-1]), self.circle(nums[1:]))
4+
5+
def circle(self, nums):
6+
"""
7+
:type nums: List[int]
8+
:rtype: int
9+
"""
10+
l = 0
11+
r = 0
12+
if len(nums) == 1:
13+
return nums[0]
14+
for num in nums:
15+
l, r = r, max(l + num, r)
16+
return r

0 commit comments

Comments
 (0)