时限1s
输入:
6 5 3 1 1 2 1 2 3 1 1 3 2 3 5 5 4 2 5 1 2
输出:
2
思路:状压dp
因为一共只有20种糖果,我们将每袋糖果种类状压成二进制存储。
枚举已知状态叠加每袋状态后的状态,O(n^2)枚举所有状态,(因为生成的状态与子状态叠加的先后无关,所以无所谓顺序,比如:状态0011 | 1100 = 1111, 于异或先后无关,“加法交换律”可证),然后维护一下新生成状态的最小值,最后看有没有符合的结果。
ps:注意初始化,注意位运算加括号,(1<<m)-1
#include <iostream>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0)
#define _for(i,a,b) for(int i=(a) ;i<=(b) ;i++)
#define _rep(i,a,b) for(int i=(a) ;i>=(b) ;i--)
#define mst(x,y) memset(x,y,sizeof(x))
#define all(v) v.begin() , v.end()
#define pb(v) push_back(v)
#define INF 0x3f3f3f3f
//#define int long long
#define lson p*2,l,mid
#define rson p*2+1,mid+1,r
//#define int long long
typedef long long ll;
const int maxn = 1<<21;
int n,m,k;
int dp[maxn];//状态i至少需要多少袋
int state[maxn];//记录状态
int cnt;//状态数
signed main()
{
//!!!!!!!!!!!!!!!!!!!!!!
// freopen("data.txt","r",stdin);
//!!!!!!!!!!!!!!!!!!!!!!
IOS;
mst(dp,0x3f);//所有状态次数初始化
cin>>n>>m>>k;
_for(i,1,n)
{
int y=0;
_for(j,1,k)
{
int x;
cin>>x;
y |= (1<<(x-1));//状压
}
dp[y] = 1;//状态i最少袋数1包
state[++cnt] = y;//存储状态
}
_for(i,1,cnt)
{
_for(j,1,maxn)
{
if( dp[j] >= INF ) continue;//某状态没出现,暂时无法算
dp[state[i]|j] = min( dp[state[i]|j] ,dp[j]+1);//维护生成状态次数最小值
}
}
if( dp[(1<<m)-1] >= INF ) cout<<-1<<endl;//不存在
else cout<<dp[(1<<m)-1]<<endl;
}