【蓝桥杯】糖果 回溯法求解
作者：互联网
【问题描述】
糖果店的老板一共有 M 种口味的糖果出售。为了方便描述，我们将 M 种口味编号 1 ∼ M。
小明希望能品尝到所有口味的糖果。遗憾的是老板并不单独出售糖果，而 是 K 颗一包整包出售。
幸好糖果包装上注明了其中 K 颗糖果的口味，所以小明可以在买之前就知道每包内的糖果口味。
给定 N 包糖果，请你计算小明最少买几包，就可以品尝到所有口味的糖果。
【输入格式】
第一行包含三个整数 N、M 和 K。
接下来 N 行每行 K 这整数 T1, T2, · · · , TK，代表一包糖果的口味。
【输出格式】
一个整数表示答案。如果小明无法品尝所有口味，输出 −1。
【样例输入】
6 5 3//N糖果包数 M口味总数 K注明的口味数；
1 1 2//糖果的口味
1 2 3
1 1 3
2 3 5
5 4 2
5 1 2
【样例输出】
2
【评测用例规模与约定】
对于 30% 的评测用例，1 ≤ N ≤ 20 。
对于所有评测样例，1 ≤ N ≤ 100，1 ≤ M ≤ 20，1 ≤ K ≤ 20，1 ≤ Ti ≤ M。
结题思路
我这里使用的简单的回溯法,当然如果会状压dp也可以,思路和01背包很像,之前写的时候想着把每一次选择的结果放到set中,这样相同的口味给只保留一个,可是最后发现,回溯的时候,没办法撤销set的这种合并,导致一开始出错,所以还是老老实实的记录选了哪几袋糖果的解向量比较靠谱.最后发现回溯法真是个万金油的解法，要好好学习。
参考代码
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
import java.util.Scanner;
class Main {
static Scanner sc = new Scanner(System.in);
static int n, m, k;
static int[][] arr;//记录每包糖果的口味
static int choice[]=new int[100];//记录哪些包糖果被选中
static int cw = 0, bestw = Integer.MAX_VALUE;//记录当前选的包数和最优选择
public static void dfs(int i) {
if (i >= n) {//解空间的树到达叶子
HashSet<Integer> all = new HashSet<Integer>();//用来方便判断是不是凑齐了所有口味
for (int l = 0; l < n; l++) {
if(choice[l]==1) {
for (int j = 0; j < k; j++) {
all.add(arr[l][j]);
}
}
}
if (all.size() == m) {
if (cw < bestw)
bestw = cw;
}
return;
}
cw++;
choice[i]=1;
dfs(i + 1);
choice[i]=0;
cw--;
//没有想剪枝条件,也可以加上提高效率
dfs(i + 1);
}
public static void main(String[] args) {
n = sc.nextInt();
m = sc.nextInt();
k = sc.nextInt();
arr = new int[n][k];
for (int i = 0; i < n; i++) {
for (int j = 0; j < k; j++) {
arr[i][j] = sc.nextInt();
}
}
dfs(0);
if (bestw == Integer.MAX_VALUE)
System.out.println(-1);
else
System.out.println(bestw);
return;
}
}
Junieson
发布了87 篇原创文章 · 获赞 527 · 访问量 9万+
私信
关注
标签：util,java,int,蓝桥,回溯,口味,import,糖果 来源： https://blog.csdn.net/i6223671/article/details/104561818