LeetCode[动态规划] - #198 House Robber

原题链接:#198 House Robber

要求:

原题是要为某盗贼设计一个能使其利益最大化的方案(这个场景并不和谐,在保持题意的情况下重新描述一个场景)。假设某糖果工厂有若干糖果机,每台糖果机每天产出不同数量的糖果,每天取糖果时不能同时取相邻两台糖果机的糖果(别问为什么),问每天能取得的最大糖果数量是多少。

糖果机产生的糖果数量集合可以看成一个整型数组。

难度:简单

分析:

考虑使用动态规划来解决此问题。当糖果机只有1台时,取走这台产出的所有糖果;当糖果机有两台时,取走产出较多的糖果。当糖果机有三台时,有两种取法,取走第二台的产出的糖果,或取出第一台与第三台的所有糖果。假设第n台糖果机产出的糖果数量为S(n), 取到第n台为止时取到的最大糖果总数量为M(n),则有:

M(1) = S(1)

M(2) = Max(S(1), S(2))

M(3) = Max(M(1)+S(3), M(2))

...

M(n) = Max(M(n-2)+S(n), M(n-1))

即是说,当取到第n台时,有两种方案,要么n-1台不取,此时糖果数量为前n-2台所取数量加上第n台的产出数量;要么第n台本身不取,保留前n-1台所取的总数量。在这两种方案中选取所取总数量较大者即可。

解决方案:

Java - 296 ms

public int rob(int[] nums) {
        if(nums==null||nums.length==0){
            return 0;
        }
        int[] maxMoney = new int[nums.length+1];
        maxMoney[nums.length] = 0;
        maxMoney[nums.length-1] = nums[nums.length-1];
        for(int i=nums.length-2; i>=0; i--){
            maxMoney[i]=(nums[i]+maxMoney[i+2]>maxMoney[i+1])?(nums[i]+maxMoney[i+2]):maxMoney[i+1];
        }
        return maxMoney[0];
    }

 简单测试程序

    

猜你喜欢

转载自cwind.iteye.com/blog/2232467
今日推荐