代码随想录算法训练营day39|动态规划part7

aaal12342024-12-12 12:06

今天是打家劫舍专题，第一题难度不大，想出状态转移方程dp[i]=max(dp[i-2]+nums[i],dp[i-1])即可求解；第二题相比于第一题，这题的房屋改成了环状结构，表面上看很难，实际上只需分类讨论，分成不包含头节点和不包含末节点两部分分别用第一题的思路求解，然后再取最大值即可。

重点是第三题，第三题是二叉树和dp的结合，难度比较大。这题的状态转移方程和前面的题目的形式都不同，因为这题是在二叉树上进行的操作，又因为操作二叉树需要采用递归，所以这道题实际上是每一层都创建一个dp数组，所以这个dp数组只需记录每一层的状态再向上返回即可。所以dp数组只需含有两个元素，一个记录不偷当前节点的状态，另一个记录偷当前节点的状态，然后逐层向上递归即可。