P8687 [蓝桥杯 2019 省 A] 糖果
~~~~~ P8687 [蓝桥杯 2019 省 A] 糖果 ~~~~~ 总题单链接
思路
~~~~~ 发现 k ≤ 20 , m ≤ 20 k\le20,m\le 20 k≤20,m≤20，考虑状压DP。
~~~~~ 预处理 g [ i ] g[i] g[i] 表示第 i i i 包糖有哪几种糖果。
~~~~~ 设 d p [ i ] dp[i] dp[i] 表示在 i i i 状态下最少要买几包糖。
~~~~~ 对于第 i i i 包糖，枚举所有状态 j j j， d p [ g [ i ] ∣ j ] = m i n ( d p [ g [ i ] ∣ j ] , d p [ j ] + 1 ) dp[g[i]|j]=min(dp[g[i]|j],dp[j]+1) 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;
}