思路如下:
求最值可以考虑采用动态规划,分析状态转移方程:
dp[i]的值应该取 dp[i-2]与nums[i]的和(因为相邻的值不能取) 或者 dp[i-1] (此时表示不打劫nums[i])之间的较大值。
class Solution {
public int rob(int[] nums) {
if(nums.length==0){
return 0;
}
if(nums.length==1){
return nums[0];
}
if(nums.length==2){
return Math.max(nums[0],nums[1]);
}
int[]dp=new int[nums.length];
dp[0]=nums[0];
dp[1]=Math.max(nums[0],nums[1]);
for(int i=2;i<nums.length;i++){
dp[i]=Math.max(dp[i-2]+nums[i],dp[i-1]);
}
return dp[nums.length-1];
}
}
原题地址:
198. 打家劫舍