Leetcode--213. 打家劫舍Ⅱ
你是一個專業(yè)的小偷,計劃偷竊沿街的房屋,每間房內(nèi)都藏有一定的現(xiàn)金。這個地方所有的房屋都圍成一圈,這意味著第一個房屋和最后一個房屋是緊挨著的。同時,相鄰的房屋裝有相互連通的防盜系統(tǒng),如果兩間相鄰的房屋在同一晚上被小偷闖入,系統(tǒng)會自動報警。
給定一個代表每個房屋存放金額的非負整數(shù)數(shù)組,計算你在不觸動警報裝置的情況下,能夠偷竊到的最高金額。
示例?1:
輸入: [2,3,2]
輸出: 3
解釋: 你不能先偷竊 1 號房屋(金額 = 2),然后偷竊 3 號房屋(金額 = 2), 因為他們是相鄰的。
示例 2:
輸入: [1,2,3,1]
輸出: 4
解釋: 你可以先偷竊 1 號房屋(金額 = 1),然后偷竊 3 號房屋(金額 = 3)。
?? ? 偷竊到的最高金額 = 1 + 3 = 4 。
?
思路:與打家劫舍那道題類似(可以參考上一篇博客),這道題其實在上一道稍微做改進即可
分兩種情況,也就是偷第一家就不偷最后一家,偷最后一家就不偷第一家,第一次是nums[0]~nums[n-2],第二次是nums[1]~nums[n-1]
兩次遍歷后觀察哪種方式收益大。
提交的代碼:
class Solution {
? ? public int rob(int[] nums) {
? ? ? ? ?int i;
?? ??? ? int n = nums.length;
?? ??? ? if(n==0)
?? ??? ? {
?? ??? ??? ? return 0;
?? ??? ? }
?? ??? ? int[] dp1 = new int[n];
?? ??? ? int[] dp2 = new int[n];
?? ??? ? if(n==1)
?? ??? ? {
?? ??? ??? ? return nums[0];
?? ??? ? }
?? ??? ? if(n==2)
?? ??? ? {
?? ??? ??? ? return java.lang.Math.max(nums[0], nums[1]);
?? ??? ? }
?? ??? ? if(n==3)
?? ??? ? {
?? ??? ??? ? return ?java.lang.Math.max(nums[0], java.lang.Math.max(nums[1],nums[2]));
?? ??? ? }
?? ??? ? dp1[0] = nums[0];
?? ??? ? dp1[1] = java.lang.Math.max(dp1[0], nums[1]);
?? ??? ? for(i=2;i<n-1;i++)
?? ??? ? {
?? ??? ??? ? dp1[i] = java.lang.Math.max(dp1[i-1], dp1[i-2]+nums[i]);
?? ??? ? } ? ??
?? ? ? ? dp2[1] = nums[1];
?? ??? ? dp2[2] = java.lang.Math.max(dp2[1], nums[2]);
?? ??? ? for(i=3;i<n;i++)
?? ??? ? {
?? ??? ??? ? dp2[i] = java.lang.Math.max(dp2[i-1], dp2[i-2]+nums[i]);
?? ??? ? }
?? ??? ? return java.lang.Math.max(dp1[n-2], dp2[n-1]);
? ? }
}
完整的代碼:
public class Solution213 {
?? ? public static int rob(int[] nums) {
?? ??? ? int i;
?? ??? ? int n = nums.length;
?? ??? ? if(n==0)
?? ??? ? {
?? ??? ??? ? return 0;
?? ??? ? }
?? ??? ? int[] dp1 = new int[n];
?? ??? ? int[] dp2 = new int[n];
?? ??? ? if(n==1)
?? ??? ? {
?? ??? ??? ? return nums[0];
?? ??? ? }
?? ??? ? if(n==2)
?? ??? ? {
?? ??? ??? ? return java.lang.Math.max(nums[0], nums[1]);
?? ??? ? }
?? ??? ? if(n==3)
?? ??? ? {
?? ??? ??? ? return ?java.lang.Math.max(nums[0], java.lang.Math.max(nums[1],nums[2]));
?? ??? ? }
?? ??? ? dp1[0] = nums[0];
?? ??? ? dp1[1] = java.lang.Math.max(dp1[0], nums[1]);
?? ??? ? for(i=2;i<n-1;i++)
?? ??? ? {
?? ??? ??? ? dp1[i] = java.lang.Math.max(dp1[i-1], dp1[i-2]+nums[i]);
?? ??? ? } ? ??
?? ? ? ? dp2[1] = nums[1];
?? ??? ? dp2[2] = java.lang.Math.max(dp2[1], nums[2]);
?? ??? ? for(i=3;i<n;i++)
?? ??? ? {
?? ??? ??? ? dp2[i] = java.lang.Math.max(dp2[i-1], dp2[i-2]+nums[i]);
?? ??? ? }
?? ??? ? return java.lang.Math.max(dp1[n-2], dp2[n-1]);
?? ? ? ?}
?? ? public static void main(String[] args)
?? ? {
?? ??? ? int[] nums = {4,1,2};
?? ??? ? System.out.println(rob(nums));
?? ? }
}
?
總結(jié)
以上是生活随笔為你收集整理的Leetcode--213. 打家劫舍Ⅱ的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Leetcode--394. 字符串解码
- 下一篇: Leetcode--461. 汉明距离