【蓝桥杯】糖果 回溯法求解

【问题描述】
糖果店的老板一共有 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;
	}

}

你可能感兴趣的:(蓝桥杯,算法)