学习记录——糖果

题目描述

糖果店的老板一共有 �M 种口味的糖果出售。为了方便描述，我们将 �M 种口味编号 1∼ �M。

小明希望能品尝到所有口味的糖果。遗憾的是老板并不单独出售糖果，而是 �K 颗一包整包出售。

幸好糖果包装上注明了其中 �K 颗糖果的口味，所以小明可以在买之前就知道每包内的糖果口味。

给定 �N 包糖果，请你计算小明最少买几包，就可以品尝到所有口味的糖果。

输入描述

第一行包含三个整数 �,�,�N,M,K。

接下来 �N 行每行 �K 个整数 �1,�2,⋅⋅⋅,��T1,T2,⋅⋅⋅,TK，代表一包糖果的口味。

其中，1≤�≤100,1≤�≤20,1≤�≤20,1≤��≤�1≤N≤100,1≤M≤20,1≤K≤20,1≤Ti≤M。

输出描述

输出一个整数表示答案。如果小明无法品尝所有口味，输出 −1。

输入输出样例

示例

输入

6 5 3
1 1 2
1 2 3
1 1 3
2 3 5
5 4 2
5 1 2

输出

2

思路：

状态压缩动态规划，d[11111]表示尝到所有糖果需要购买的最小糖果包数，a[i]=00110表示第i包糖果中含有第二种糖果和第三种糖果。

状态转移方程：d[j]=min(d[j],d[i]+1)，其中的j=i|a[k](k=0,1,2,3.....)。

做法：

#include<iostream>
#include<string.h>
using namespace std;
const int M=(1<<20)+10;
int d[M];
int main(){
    int n,m,k;
    int x;
    int a[101];
    cin>>n>>m>>k;
    memset(d,-1,sizeof(d));
    for(int i=0;i<n;i++){
        int s=0;
        for(int j=0;j<k;j++){
            cin>>x;
            s=s|(1<<(x-1));
        }
    a[i]=s;
    d[s]=1;
    }
    d[0]=0;
//***************************
    for(int i=0;i<n;i++){
        for(int j=0;j<=(1<<m);j++){
        if(d[j]==-1) continue;
        int to=j|a[i];
        if(d[to]!=-1) d[to]=min(d[to],d[j]+1);
        else d[to]=d[j]+1;
        }
    }
//**************************
    printf("%d",d[(1<<m)-1]);
    return 0;
}