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