AcWing 1064. 小国王
原题链接
简单
作者:
leimingze
,
2022-04-01 23:19:19
,
所有人可见
,
阅读 133
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=12;
const int M=1<<10,K=110;
vector<int>state;
vector<int>head[M];
int cnt[M];
LL f[N][K][M];
int n,m;
bool check(int state)
{
for(int i=0;i<n;i++)
{
if((state>>i&1)&&(state>>i+1&1))
return false;
}
return true;
}
int count(int state)
{
int res=0;
for(int i=0;i<n;i++)
if(state>>i&1)res++;
return res;
}
int main()
{
cin>>n>>m;
for(int i=0;i<1<<n;i++)
{
if(check(i))
{
state.push_back(i);
cnt[i]=count(i);
}
}
f[0][0][0]=1;
for(int i=0;i<state.size();i++)
{
for(int j=0;j<state.size();j++)
{
int a=state[i],b=state[j];
if((a&b)==0&&check(a|b))
head[i].push_back(j);
}
}
//状态压缩+01背包
for(int i=1;i<=n+1;i++)
{
for(int j=0;j<=m;j++)
{
for(int a=0;a<state.size();a++)
{
for(auto b:head[a])
{
int c=cnt[state[a]];
if(j>=c)f[i][j][a]+=f[i-1][j-c][b];
}
}
}
}
cout<<f[n+1][m][0]<<endl;
return 0;
}