【蓝桥杯】神盾神与砝码称重——DFS(深度优先算法)
一、递归参数:
指向针,标记当前位置、累加器:存储当前重量
二、递归出口:
①当现在加的砝码,和物体的重量相等的时候,可以称出重量,递归结束;
②如果当现在加的砝码个数已经达到上限了(没有砝码可以用了),称不出来,单次子递归调用结束
三、递归规律:
每一层调用递归的时候,都比上一层多使用了一个砝码,而这个砝码可以选择不加,直接扔掉,或者可以加在物品一端,或者加在物品的另外一端
采用深度优先搜索,此代码多数是借鉴,正在学习,每个题都是不一样的解法,但根本一致,即深搜。应重思想。
static int n;//记录砝码个数
static int t;//记录单个物品的重量
static boolean flag = false;//记录是否可以称出
static int[] weights;//砝码
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
n = sc.nextInt(); //输入,砝码的个数
int m = sc.nextInt(); //输入,物品的个数:
weights = new int[n];
for (int i = 0; i < n; i++) { //输入,每个砝码的重量
weights[i] = sc.nextInt();
}
for (int i = 0; i < m; i++) {
t = sc.nextInt(); //输入,物品的重量
//递归开始,index为指针从0开始,sum和从0开始
dfs(0, 0);
if (flag) {
System.out.println("YES");
} else
System.out.println("NO");
}
}
public static void dfs(int index, int sum) {
//如果物品t = 目前砝码端重量,表示可以称出
if (t == sum) {
flag = true;
return;
}
//如果砝码个数已用完,单词子递归结束
if (index == n) {
return;
}
//砝码放在左天平、右天平、不放砝码
dfs(index + 1, sum + weights[index]);
dfs(index, sum);
dfs(index - 1, sum - weights[index]);
}