Update 0213.打家劫舍II.md

增加python - 二维dp数组写法
This commit is contained in:
ZerenZhang2022
2023-04-10 00:58:56 -04:00
committed by GitHub
parent ba03a17e68
commit 34aabac5b9

View File

@ -142,7 +142,20 @@ class Solution:
dp[i]=max(dp[i-1],dp[i-2]+nums[i])
return dp[-1]
```
```python
class Solution: # 二维dp数组写法
def rob(self, nums: List[int]) -> int:
if len(nums)<3: return max(nums)
return max(self.default(nums[:-1]),self.default(nums[1:]))
def default(self,nums):
dp = [[0,0] for _ in range(len(nums))]
dp[0][1] = nums[0]
for i in range(1,len(nums)):
dp[i][0] = max(dp[i-1])
dp[i][1] = dp[i-1][0] + nums[i]
return max(dp[-1])
```
Go
```go