蓝桥 糖果 (状压dp)

题目描述
糖果店的老板一共有 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

对于每一包 有第i种糖果就标记为1,没有的话就是0

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
//dp[i]在i种状态下所需要最少包数
int candy[105];
int dp[(1<<20)+5];
int main(){
	int n, m, k;
	cin>>n>>m>>k;
	memset(dp, -1, sizeof(dp));
	for (int i=1; i<=n; i++){
		for (int j=1; j<=k; j++){
			int t;
			scanf("%d", &t);
			candy[i] |= (1<<(t-1)); 
		}
		dp[candy[i]]=1;
	}
	for (int i=1; i<=n; i++){
		for (int j=0; j<(1<<m); j++){
			if (dp[j]==-1)	continue;
			if(dp[j|candy[i]]==-1)	dp[j|candy[i]]=dp[j]+1;
			else	dp[j|candy[i]]=min(dp[j|candy[i]], dp[j]+1); 
		}
	}
	printf("%d\n", dp[(1<<m)-1]);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_42172261/article/details/105983110