leetcode算法题--打家劫舍

原题链接:https://leetcode-cn.com/problems/house-robber-ii/
用动态规划来做,难点在于是环状的,解决办法是分别计算去掉首和尾的版本,目前没有更好的方法
动态转移方程:

dp[i]=std::max(dp[i-1],dp[i-2]+nums[i])

代码:

	int rob(vector<int>& nums) {
        int len=nums.size();
        if(len==0) return 0;
        if(len==1) return nums[0];
        vector<int> vec1(nums.begin(),nums.end()-1);
        vector<int> vec2(nums.begin()+1,nums.end());
        return std::max(func(vec1),func(vec2));
    }
    int func(vector<int> &nums){
        int len=nums.size();
        if(len==1) return nums[0];
        vector<int> dp(len,0);
        dp[0]=nums[0];
        dp[1]=max(nums[0],nums[1]);
        for(int i=2;i<len;i++)
            dp[i]=std::max(dp[i-1],dp[i-2]+nums[i]);
        return dp[len-1];
     }
发布了75 篇原创文章 · 获赞 12 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/qq_20817327/article/details/105008158
今日推荐