动态规划强盗抢劫Java实现

题目:抢劫一排住户,但是不能抢邻近的住户,求最大抢劫量

思路:dp(n)代表抢劫到第n个住户的最大抢劫量,如果抢劫了第n-1个住户,就不能抢劫第n个住户,用arr数组代表一排用户

所以dp(n)=max(dp(n-2)+arr[n],dp(n-1))

//动态规划,强盗抢劫
public class Rob {
    public static int rob(int[] arr){
        int pre=0;
        int cur=0;//用来记录抢劫到第i个住户的最大抢劫量
        for(int i=0;i<arr.length;i++){
            int next=Math.max(pre+arr[i],cur);
            pre=cur;
            cur=next;
        }
        return cur;
    }
    public static void main(String[] args){
        int[] arr={1,2,6,8,9};
        int[] arr2={1,3,7,5,0};
        System.out.println(rob(arr));//16
        System.out.println(rob(arr2));//8

    }
}

猜你喜欢

转载自blog.csdn.net/matrixZCL/article/details/90257079