P8687 [蓝桥杯 2019 省 A] 糖果

      ~~~~~      P8687 [蓝桥杯 2019 省 A] 糖果       ~~~~~      总题单链接

思路

      ~~~~~      发现 $k\le 20,m\le 20$，考虑状压DP。

      ~~~~~      预处理 $g[i]$ 表示第 $i$ 包糖有哪几种糖果。

      ~~~~~      设 $dp[i]$ 表示在 $i$ 状态下最少要买几包糖。

      ~~~~~      对于第 $i$ 包糖，枚举所有状态 $j$，$dp[g[i]|j]=min(dp[g[i]|j],dp[j]+1)$。

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;

ll n,m,Q,g[105],dp[(1ll<<20)+5];

signed main(){
	ios::sync_with_stdio(false);
	
	cin>>n>>m>>Q;
	for(ll i=1;i<=n;i++){
		for(ll j=1;j<=Q;j++){
			ll k;cin>>k;
			g[i]|=(1ll<<(k-1));
		}
	}
	
	memset(dp,0x3f,sizeof(dp));
	dp[0]=0;
	
	for(ll i=1;i<=n;i++)
		for(ll j=(1ll<<m)-1;j>=0;j--)
			dp[g[i]|j]=min(dp[g[i]|j],dp[j]+1);
	
	cout<<(dp[(1ll<<m)-1]<=n?dp[(1ll<<m)-1]:-1);
	
	return 0;
}