From ede276ea268dc4b99cee85c2c401bfb880d34b60 Mon Sep 17 00:00:00 2001 From: Gabriel Donnan <47415809+gabedonnan@users.noreply.github.com> Date: Thu, 12 Jan 2023 10:59:25 +0000 Subject: [PATCH] Create house-robber-II.py Same as previous house robber solution but houses are arranged circularly --- house-robber-II.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 house-robber-II.py diff --git a/house-robber-II.py b/house-robber-II.py new file mode 100644 index 0000000..8919d59 --- /dev/null +++ b/house-robber-II.py @@ -0,0 +1,16 @@ +class Solution(object): + def rob(self, nums): + return max(nums[0] + self.circle(nums[2:-1]), self.circle(nums[1:])) + + def circle(self, nums): + """ + :type nums: List[int] + :rtype: int + """ + l = 0 + r = 0 + if len(nums) == 1: + return nums[0] + for num in nums: + l, r = r, max(l + num, r) + return r