打家劫舍II

Posted 2022-08-12 guohai-stronger

题目描述(LeetCode)

你是一个专业的小偷，计划偷窃沿街的房屋，每间房内都藏有一定的现金。这个地方所有的房屋都围成一圈，这意味着第一个房屋和最后一个房屋是紧挨着的。同时，相邻的房屋装有相互连通的防盗系统，如果两间相邻的房屋在同一晚上被小偷闯入，系统会自动报警。

给定一个代表每个房屋存放金额的非负整数数组，计算你在不触动警报装置的情况下，能够偷窃到的最高金额。

示例 1:

输入: [2,3,2]
输出: 3
解释: 你不能先偷窃 1 号房屋（金额 = 2），然后偷窃 3 号房屋（金额 = 2）, 因为他们是相邻的。

示例 2:

输入: [1,2,3,1]
输出: 4
解释: 你可以先偷窃 1 号房屋（金额 = 1），然后偷窃 3 号房屋（金额 = 3）。
     偷窃到的最高金额 = 1 + 3 = 4 。

 

题目讲解

打家劫舍 II 和 打家劫舍 相比，题目只有一个变化。
这个地方所有的房屋都围成一圈，这意味着第一个房屋和最后一个房屋是紧挨着的。
这个规则会带来怎样的影响呢？
如果按照原来的解法，最要命的问题就是无法确定是否即抢了第一家又抢了最后一家。
那么，要怎么保证抢了第一家就不抢最后一家呢？

 

对于上面我们可以采取拆解的方式,变为两个打家劫舍I的方式,如下:

相比于第一题，此题目将首位也作为相邻，假设总共有N个房子，思路是：
1.分两次求解，首先是第一家到第N-1家，求得偷得最多的值，记Max01;
2.之后再次求得第2家到第N家可以偷的最大值，记为Max02；
3.最后比较Max01与Max02，将最大的那个值返回。

 

题目代码

对于上面的逻辑，用swift代码如下：

import UIKit

let numsArray: [Int] = [1,2,3,1]
func robs(numsArr: [Int]) -> Int 
    if numsArr.count == 0 
        return 0
    
    if numsArr.count == 1 
        return numsArr[0]
    
    if numsArr.count == 2 
        if numsArr[0] < numsArr[1] 
            return numsArr[1]
         else 
            return numsArr[0]
        
    
    var dp1 = [Int]()
    var dp2 = [Int]()
    /**
     *偷第一家与不偷第一家
     */
    
    //偷第一家
    for i in 0..<numsArr.count - 1 
        if i < 2 
            dp1.append(numsArr[0])
            if dp1[0] < numsArr[1] 
                dp1.append(numsArr[1])
             else 
                dp1.append(dp1[0])
            
         else 
            let fk_1 = dp1[i - 1]
            let fk_2 = dp1[i - 2] + numsArr[i - 1]
            let temp = fk_1 > fk_2 ? fk_1 : fk_2
            dp1.append(temp)
        
    
    
    //不偷第一家，可以偷最后一家
    for i in 1..<numsArr.count 
        if i < 3 
            dp2.append(numsArr[1])
            if dp2[0] < numsArr[2] 
                dp2.append(numsArr[2])
             else 
                dp2.append(dp2[0])
            
         else 
            let fk_1 = dp2[i - 1]
            let fk_2 = dp2[i - 2] + numsArr[i]
            let temp = fk_1 > fk_2 ? fk_1 : fk_2
            dp2.append(temp)
        
    
    
    //比较大小
    let maxRobs = dp1[dp1.count - 1] > dp2[dp2.count - 1] ? dp1[dp1.count - 1] : dp2[dp2.count - 1]
    return maxRobs


let maxRobs = robs(numsArr: numsArray)
print(maxRobs)

通过playground打印出结果如下：

 直接拷贝上面代码即可！

 

上面就是打家劫舍II的版本，希望对大家理解有所帮助，看完麻烦点个赞呗，谢谢！