天天看點

JAVA程式設計:打家劫舍 II(LeetCode:213)

你是一個專業的小偷,計劃偷竊沿街的房屋,每間房内都藏有一定的現金。這個地方所有的房屋都圍成一圈,這意味着第一個房屋和最後一個房屋是緊挨着的。同時,相鄰的房屋裝有互相連通的防盜系統,如果兩間相鄰的房屋在同一晚上被小偷闖入,系統會自動報警。

給定一個代表每個房屋存放金額的非負整數數組,計算你在不觸動警報裝置的情況下,能夠偷竊到的最高金額。

示例 1:

輸入: [2,3,2]

輸出: 3

解釋: 你不能先偷竊 1 号房屋(金額 = 2),然後偷竊 3 号房屋(金額 = 2), 因為他們是相鄰的。

示例 2:

輸入: [1,2,3,1]

輸出: 4

解釋: 你可以先偷竊 1 号房屋(金額 = 1),然後偷竊 3 号房屋(金額 = 3)。

     偷竊到的最高金額 = 1 + 3 = 4 。

思路:如果這道題不是環的話是個史詩級水的動态規劃題,但是這題是環,好吧,其實還是水題,我們隻需要暴力枚舉第一個東西偷還是不偷就行了,如果偷了,呢最後一個就一定不能偷,其他地方直接按照正常dp即可。

dp[i][0]:表示第i個不偷的目前最大值。

dp[i][1]:表示偷第i個時的目前最大值

轉移方程詳見代碼:

class Solution {
    public int rob(int[] nums) {
    	if(nums.length==0)
    		return 0;
    	
    	int ans=0;
        int[][] dp=new int[nums.length][2];
        
        for(int i=0;i<nums.length;i++)
        {
        	if(i==0)
        		continue;
        	
        	dp[i][0]=Math.max(dp[i-1][0], dp[i-1][1]);
        	dp[i][1]=dp[i-1][0]+nums[i];
        }
        
        ans=Math.max(dp[nums.length-1][0], dp[nums.length-1][1]);
        
        for(int i=0;i<nums.length;i++)
        	for(int j=0;j<2;j++)
        		dp[i][j]=0;
        
        for(int i=0;i<nums.length;i++)
        {
        	if(i==0)
        	{
        		dp[i][1]=nums[i];
        		continue;
        	}
        	
        	dp[i][0]=Math.max(dp[i-1][0], dp[i-1][1]);
        	dp[i][1]=dp[i-1][0]+nums[i];
        }
        
        ans=Math.max(ans, nums.length==1?nums[0]:dp[nums.length-1][0]);
        
        return ans;
    }
}
           

繼續閱讀