动态规划入门问题
我是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));
}