题解 | #打家劫舍(一)#

打家劫舍(一)

https://www.nowcoder.com/practice/c5fbf7325fbd4c0ea3d0c3ea6bc6cc79?tpId=295&tqId=2285793&ru=/exam/oj&qru=/ta/format-top101/question-ranking&sourceUrl=%2Fexam%2Foj

C语言写吐了,不想写了。。。

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型
     */
    int rob(vector<int>& nums) {
      //  dp[i] 表示偷至前i家时所窃取的最多金钱
      int size = nums.size();
      std::vector<int> dp(size + 1, 0);
      dp[0] = 0;
      dp[1] = nums[0];
      
      for (int i = 2; i <= size; ++i) {
        dp[i] = std::max(dp[i - 1], dp[i - 2] + nums[i - 1]);
      }
      
      return dp[size];
    }
};

跟跳台阶一样,空间复杂度可以降为1

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型
     */
    int rob(vector<int>& nums) {
      int a = 0, b = nums[0], c = nums[0];
      
      for (int i = 2; i <= nums.size(); ++i) {
        c = std::max(b, a + nums[i - 1]);
        a = b;
        b = c;
      }
      
      return c;
    }
};
全部评论

相关推荐

11-04 19:05
已编辑
东莞城市学院 单片机
不知道怎么取名字_:你这个要实习两年?哪有这么久的,感觉就是即使你毕业了,但还按实习的话,是不是不用给你缴社保公积金啥的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务