当前位置:   article > 正文

打家劫舍I & 打家劫舍II (leetcode)

打家劫舍I & 打家劫舍II (leetcode)

个人主页:Lei宝啊 

愿所有美好如期而遇


打家劫舍Iicon-default.png?t=N7T8https://leetcode.cn/problems/Gu0c2T/打家劫舍IIicon-default.png?t=N7T8https://leetcode.cn/problems/PzWKhm/

状态转移方程就是这样的:

  • i位置选择偷f[i]:f[i] = g[i-1] + nums[i];
  • i位置选择不偷g[i]:g[i] = max(f[i-1], g[i-1]);  
  1. class Solution
  2. {
  3. public:
  4. int rob(vector<int>& nums)
  5. {
  6. int num = nums.size();
  7. if(num == 0) return 0;
  8. vector<int> g(num), f(num);
  9. f[0] = nums[0], g[0] = 0;
  10. for(int i=1; i<num; i++)
  11. {
  12. f[i] = g[i-1] + nums[i];
  13. g[i] = max(f[i-1], g[i-1]);
  14. }
  15. return max(f[num-1], g[num-1]);
  16. }
  17. };

 

  1. class Solution
  2. {
  3. public:
  4. int massage(int lhs, int rhs, vector<int>& nums)
  5. {
  6. if(lhs > rhs) return 0;
  7. vector<int> g(nums.size()), f(nums.size());
  8. //这里不需要初始化f[lhs],因为f[i]的状态转移方程不会越界
  9. //而上面的f[0]需要初始化是因为f[0]的状态转移方程会越界,所以从1开始
  10. for(int i=lhs; i<=rhs; i++)
  11. {
  12. f[i] = g[i-1] + nums[i];
  13. g[i] = max(f[i-1], g[i-1]);
  14. }
  15. return max(f[rhs], g[rhs]);
  16. }
  17. int rob(vector<int>& nums)
  18. {
  19. int n = nums.size();
  20. //偷
  21. int lhs = massage(2, n-2, nums) + nums[0];
  22. //不偷
  23. int rhs = massage(1, n-1, nums);
  24. return max(lhs, rhs);
  25. }
  26. };

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Monodyee/article/detail/655973
推荐阅读
相关标签
  

闽ICP备14008679号