198.本题即在给定的数组arr中,求出arr中不相邻节点之和最大值。
与之前写的动态规划那题一样。我试着用递归和非递归的方法去写。递归的方法,时间复杂度超了,非递归的通过。
代码:
class Solution:
def rob(self, nums: List[int]) -> int:
#动态规划,本题求出arr中不相邻节点之和最大值
#解题思路:
#方法一:递归
if not nums:
return 0
if len(nums) == 1:
return nums[0]
if len(nums) == 2:
return max(nums[0],nums[1])
# def rec_opt(arr,i):
# if i == 0:
# return arr[0]
# elif i == 1:
# return max(arr[0],arr[1])
# else:
# A = rec_opt(arr,i-2) + arr[i]
# B = rec_opt(arr,i-1)
# return max(A,B)
# return rec_opt(nums,len(nums)-1)
#方法二:非递归方法
import numpy as np
def dp_opt(arr):
opt = np.zeros(len(arr))
#下标0最优解就是arr[0]
#下标1最优解是max(arr[0], arr[1])
opt[0] = arr[0]
opt[1] = max(arr[0], arr[1])
for i in range(2,len(arr)):
A = opt[i-2] + arr[i]
B = opt[i-1]
opt[i] = max(A, B)
return opt[len(arr)-1]
result = dp_opt(nums)
return int(result)
213.这题在198题基础之上把所有的房屋都围成一圈
思路:
- 第一个选,最后一个就不能选
- 最后一个选,第一个就不能选
- 求出以上两种方案的最大值
class Solution:
def rob(self, nums: List[int]) -> int:
if not nums:
return 0
if len(nums) == 1:
return nums[0]
if len(nums) == 2:
return max(nums[0],nums[1])
#方法二:非递归方法
import numpy as np
def dp_opt(arr):
opt = np.zeros(len(arr))
#下标0最优解就是arr[0]
#下标1最优解是max(arr[0], arr[1])
opt[0] = arr[0]
opt[1] = max(arr[0], arr[1])
for i in range(2,len(arr)):
A = opt[i-2] + arr[i]
B = opt[i-1]
opt[i] = max(A, B)
return opt[len(arr)-1]
result = dp_opt(nums[0:len(nums)-1])
result2 = dp_opt(nums[1:len(nums)])
return int(max(result,result2))
- 打家劫舍 III
输入: [3,2,3,null,3,null,1]
3
/ \
2 3
\ \
3 1
输出: 7
解释: 小偷一晚能够盗取的最高金额 = 3 + 3 + 1 = 7.
思路
- 用列表存储不偷和偷的状态
[不偷,偷] - 不偷= max左树 + max右树
- 偷 = 当前节点(n.val) + 左树不偷+右树不偷
class Solution:
def rob(self, root: TreeNode) -> int:
def hit(node):
if not node:
return [0,0]
res = [0,0]
l = hit(node.left)
r = hit(node.right)
res[0] = max(l)+max(r)
res[1] = node.val + l[0] +r[0]
return res
return max(hit(root))