牛客题目——打家劫舍一、打家劫舍二


题目1——打家劫舍一

你是一个经验丰富的小偷,准备偷沿街的一排房间,每个房间都存有一定的现金,为了防止被发现,你不能偷相邻的两家,即如果偷了第一家,就不能再偷第二家;如果偷了第二家,那么就不能偷第一家和第三家。
给定一个整数数组nums,数组中的元素表示每个房间存有的现金余额,请你计算在不被发现的前提下最多的偷窃金额。

示例
输入:[1,2,3,4]
输出:6(最优方案是偷第2、4个房间)

解题思路

贪心算法和动态规划的共同点是:两者都具有最优子结构性质,即问题的最优解中包含子问题的最优解。
两者的不同点是:
动态规划中,每步做出的选择往往依赖于相关子问题的解,因而只有在解除相关子问题时才能做出选择。
而贪心算法中,仅在当前状态下做出最好选择,然后再去解这个选择做出后产生的子问题。

如果利用贪心算法偷钱,为了偷取更多的钱,可能会连续放弃两家不偷,因此这种方法不可行,考虑利用动态规划来解决。对于每一个人家,可选择偷或者不偷,状态转移方程为dp[i] = Math.max(dp[i-1], dp[i-2]+nums[i-1)

代码实现

import java.util.*;
public class Solution {
    public int rob (int[] nums) {
        int n = nums.length;
        if(n == 1)
            return nums[0];
        int[] dp = new int[n+1];
        dp[1] = nums[0];
        for(int i=2;i<=n;i++){
            //当偷到第i个房间时,可以选择偷或者不偷
            //偷的话,则累加上上个收益,否则累计的为上级的收益
            dp[i] = Math.max(dp[i-1],dp[i-2]+nums[i-1]);
        }
        return dp[n];
    }
}

题目2——打家劫舍二

你是一个经验丰富的小偷,准备偷沿湖的一排房间,每个房间都存有一定的现金,为了防止被发现,你不能偷相邻的两家,即偷了第一家,就不能偷第二家,如果偷了第二家,那么就不能偷第一家和第三家。沿湖的房间组成一个闭合的圆形,即第一个房间和最后一个房间视为相邻。
给定一个长度为n的整数数组nums,数组中的元素表示每个房间存有的现金数额,请你计算在不被发现的前提下最多的偷窃金额。

示例
输入:[1,3,6]
输出:6(1和3相邻,因此最优方案是偷第3个房间)

解题思路

与题目一类似,只是这道题是环形,第一家和最后一家相邻,不能同时取到,所以可分为两种情况:偷第一家,则最后一家不能偷;偷最后一家,则第一家不能偷。除去考虑这些,其他家与第一题做法相同。
最后取两种情况中最大的值即可。

代码实现

import java.util.*;
public class Solution {
    public int rob (int[] nums) {
        int n = nums.length;
        if(n == 1)
            return nums[0];
        int[] dp = new int[n+1];
        //偷第一家
        dp[1] = nums[0];
        for(int i=2;i<n;i++){
            dp[i] = Math.max(dp[i-1], dp[i-2]+nums[i-1]);
        }
        int res = dp[n-1];
        Arrays.fill(dp,0);
        //偷最后一家
        for(int i=2;i<=n;i++){
            dp[i] = Math.max(dp[i-1],dp[i-2]+nums[i-1]);
        }
        return Math.max(res,dp[n]);
    }
}
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值