动态规划入门问题

动态规划入门问题

我是B站搬运工（手动狗头），参考视频是灯神的动态规划动态规划 (第1讲)_哔哩哔哩_bilibili，动态规划（第2讲）_哔哩哔哩_bilibili，本文实现了第二个视频中两个题目的java代码。

一、在一个数组arr中，找出一组不相邻的数字，使得最后的和最大。

• 首先是递归方法

	public static int recOpt(int[] arr, int i){
        if(i == 0){
            return arr[0];
        } else if(i == 1){
            return Math.max(arr[0],arr[1]);
        } else{
            return Math.max(recOpt(arr,i - 2) + arr[i], recOpt(arr,i - 1));
        }
    }

• 然后是DP

	public static int dpOpt(int[] arr){
        int[] opt = new int[arr.length];
        opt[0] = arr[0];
        opt[1] = Math.max(arr[0],arr[1]);
        for (int i = 2; i < arr.length; i++){
            int A = opt[i - 2] + arr[i];
            int B = opt[i - 1];
            opt[i] = Math.max(A,B);
        }
        return opt[arr.length - 1];
    }

• main方法测试

	public static void main(String[] args) {
        int[] arr = {1,2,4,1,7,8,3};
        System.out.println(recOpt(arr, arr.length - 1));
        System.out.println(dpOpt(arr));
    }

二、给定一个正整数s, 判断一个数组arr中，是否有一组数字加起来等于s。

• 首先是递归方法

	public static boolean recSubset(int[] arr, int i, int s) {
        if (s == 0) {
            return true;//匹配完毕目标值
        } else if (i == 0) {
            return arr[0] == s;//到达数组第一个数值，判断是否等于s
        } else if (arr[i] > s) {
            return recSubset(arr, i - 1, s);//不选这个分支，因为加上arr[i]已经大于目标值，直接选择另一个分支
        } else {
            boolean A = recSubset(arr, i - 1, s - arr[i]);//选
            boolean B = recSubset(arr, i - 1, s);//不选
            if(A){
                return A;
            }
            if(B){
                return B;
            }
        }
        return false;
    }

• 然后是DP

	public static boolean dpSubset(int[] arr,int s){
        boolean[][] subset = new boolean[arr.length][s + 1];//构建数组存储得到的数据
        //第一行全部置为false
        for (int i = 0; i < s + 1; i++){
            subset[0][i] = false;
        }
        //第一列全部置为true，这个要比第一行置为false靠后，因为数组[0][0]位置得是true
        for (int i = 0; i < arr.length; i++){
            subset[i][0] = true;
        }
        //如果arr第一个元素和s剩余值相同，将那个位置置为true
        if (arr[0] <= s) {//防止数组越界，提前判断
            subset[0][arr[0]] = true;
        }

        for (int i = 1; i < arr.length; i++){
            for (int j = 1; j < s + 1; j++){//j为s剩余值

                if (arr[i] > j){//该值超过s剩余值，不走该分支
                    subset[i][j] = subset[i - 1][j];
                }else{
                    boolean A = subset[i - 1][j - arr[i]];//选
                    boolean B = subset[i - 1][j];//不选
                    subset[i][j] = A || B;
                }

            }
        }
        return subset[arr.length - 1][s];
    }

• main方法测试

	public static void main(String[] args) {
        int[] arr = {3, 34, 4, 12, 5, 2};
        System.out.println(recSubset(arr, arr.length - 1, 9));
        System.out.println(recSubset(arr, arr.length - 1, 10));
        System.out.println(recSubset(arr, arr.length - 1, 11));
        System.out.println(recSubset(arr, arr.length - 1, 12));
        System.out.println(recSubset(arr, arr.length - 1, 13));
        System.out.println(dpSubset(arr,9));
        System.out.println(dpSubset(arr,10));
        System.out.println(dpSubset(arr,11));
        System.out.println(dpSubset(arr,12));
        System.out.println(dpSubset(arr,13));
    }