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;
}