题解 | 打家劫舍(二)

打家劫舍(二)

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

#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param nums int整型vector 
     * @return int整型
     */
    int rob(vector<int>& nums) {
        if(nums.empty()) return 0; // 处理空数组情况
        if(nums.size()==1) return nums[0];

        int n=nums.size();
        vector<int> dp1(n,0);
        vector<int> dp2(n,0);

        dp1[0]=nums[0];
        dp1[1]=max(nums[0],nums[1]);
        dp2[0]=0;
        dp2[1]=nums[1];

        for(int i=2;i<n-1;i++)
        {
            dp1[i]=max(dp1[i-1],dp1[i-2]+nums[i]);
        }

        for(int i=2;i<n;i++)
        {
            dp2[i]=max(dp2[i-1],dp2[i-2]+nums[i]);
        }
        return max(dp1[n-2],dp2[n-1]);
    }
};

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

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