P8687 [蓝桥杯 2019 省 A] 糖果

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

思路

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

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

~~~~~ 设 d p i dpi dpi 表示在 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 ) dpg\[i|j]=min(dpg\[i|j],dpj+1) dpg\[i∣j]=min(dpg\[i∣j],dpj+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;
}