动态规划强盗抢劫Java实现

原创于 2019-05-16 09:47:46 发布 · 805 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

13 篇文章

订阅专栏

本文介绍了一种使用动态规划算法解决特殊抢劫问题的方法，即在不连续抢相邻住户的条件下，求得最大抢劫金额。通过定义状态转移方程dp(n)，展示了如何递推计算至最优解，提供了一个具体实现的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

    }
}