【Lintcode】392. House Robber

題目地址:

https://www.lintcode.com/problem/house-robber/description

給定一個數組AA,每個位置代表這個位置的財富值。現允許取任意多位置的財富,但不允許取相鄰的。問最多能取多少財富。

思路是動態規劃。設f[i]f[i]表示前ii個位置(從00開始取)能取到的最大的財富值。那麼f[0]f[0]對應的就是什麼也不取,所以f[0]=0f[0]=0。對於位置ii,有兩種情況,若取,則能得到的最大財富值爲A[i]+f[i2]A[i]+f[i-2];若不取,則能得到的最大財富值爲f[i1]f[i-1]。所以f[i]=max{A[i]+f[i2],f[i1]}f[i]=\max\{A[i]+f[i-2], f[i-1]\}代碼如下:

public class Solution {
    /**
     * @param A: An array of non-negative integers
     * @return: The maximum amount of money you can rob tonight
     */
    public long houseRobber(int[] A) {
        // write your code here
        if (A == null || A.length == 0) {
            return 0;
        }
        
        if (A.length == 1) {
            return A[0];
        }
        
        long[] dp = new long[A.length + 1];
        dp[1] = A[0];
        for (int i = 2; i <= A.length; i++) {
            dp[i] = Math.max(dp[i - 1], A[i - 1] + dp[i - 2]);
        }
        
        return dp[A.length];
    }
}

時空複雜度O(n)O(n)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章