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

打家劫舍(二)

http://www.nowcoder.com/practice/a5c127769dd74a63ada7bff37d9c5815

动态规划

相同的代码跑两遍,和(一)思路一样

f[i] 表示偷第 i 个房间,g[i] 表示不偷第 i 个房间

这里分两种情况:

  1. 不选起点,答案就是 res = max(f[n], g[n]) 最大值

  2. 选起点,那么就不能选终点,所以答案就是 res = max(res, g[n])

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型
     */
    int rob(vector<int>& nums) {
        int n = nums.size();
        vector<int> f(n + 1), g(n + 1);
        
        // 1. 不选起点
        f[0] = 0, f[1] = 0;
        g[0] = 0, g[1] = 0;
        for (int i = 2; i <= n; i ++ ) {
            f[i] = g[i - 1] + nums[i - 1];
            // 如果不选 i,那么 i - 1 可选可不选
            g[i] = max(f[i - 1], g[i - 1]);
        }
        
        int res = max(f[n], g[n]);
        
        // 2. 选起点,那么就不能选终点
        f[0] = 0, f[1] = nums[0];
        g[0] = 0, g[1] = 0;
        for (int i = 2; i <= n; i ++ ) {
            f[i] = g[i - 1] + nums[i - 1];
            g[i] = max(f[i - 1], g[i - 1]);
        }
        
        return max(res, g[n]);
    }
};
全部评论

相关推荐

今天 12:22
门头沟学院 Java
点赞 评论 收藏
分享
06-12 16:23
已编辑
小米_软件开发(准入职员工)
点赞 评论 收藏
分享
07-08 13:48
门头沟学院 C++
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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