回溯

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

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值