题目描述
糖果店的老板一共有 M 种口味的糖果出售。为了方便描述,我们将 M 种 口味编号 1 ∼ M。
小明希望能品尝到所有口味的糖果。遗憾的是老板并不单独出售糖果,而 是 K 颗一包整包出售。
幸好糖果包装上注明了其中 K 颗糖果的口味,所以小明可以在买之前就知 道每包内的糖果口味。
给定 N 包糖果,请你计算小明最少买几包,就可以品尝到所有口味的糖 果。
输入
第一行包含三个整数 N、M 和 K。
接下来 N 行每行 K 这整数 T1, T2, · · · , TK,代表一包糖果的口
输出
一个整数表示答案。如果小明无法品尝所有口味,输出 −1。
样例输入
6 5 3
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。
思路:
- M为20小数量级,状态dp题型
- 把每包糖果看作集合,用整数来表示这一集合
- 每袋糖果的种类用二进制(整数)表示,含有第 i 种糖果记为第 i 位为1
- 用已有的状态去更新加上新一包糖果后的状态
- 首先dp在某状态遇到-1,则跳过;dp在某状态加上已拿取得sta状态为-1,或者更新状态后包数比以前的更少,则加上新的并更新dp+1.
程序:
import java.util.Arrays;
import java.util.Scanner;
public class Main {
static Scanner sc=new Scanner(System.in);
static int dp[];
static int sta[];
public static void main(String[] args) {
int n = sc.nextInt();
int m = sc.nextInt();
int k = sc.nextInt();
dp = new int[(1 << 20) + 5];
sta = new int[n + 10];
Arrays.fill(dp, -1);
for (int i = 0; i < n; i++) {
for (int j = 0; j < k; j++) {
int a = sc.nextInt();
sta[i] |= 1 << (a - 1); //状态压缩
}
dp[sta[i]] = 1; //这个状态在中dp的状态为 1,(选一次)
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < (1 << m); j++) { //sta 种类最多为1<<m
if (dp[j] == -1) continue;
if (dp[j | sta[i]] == -1 || dp[j] + 1 < dp[j | sta[i]]) {
dp[j | sta[i]] = dp[j] + 1;
}
}
}
System.out.println(dp[(1 << m) - 1]);
}
}
另一种状态转移方程(易于理解)
if(dp[j]==-1) continue;
else if(dp[j|sta[i]]==-1) dp[j|sta[i]]=dp[j]+dp[sta[i]];
else dp[j|sta[i]]=min(dp[j|sta[i]],dp[j]+dp[sta[i]]);