这个题动态转移方程式不难想,用dp[i][j]表示前i头牛用状态j满足条件,其中j的每一位表示一个barns,然后转移就很简单了,dp[i][j|(1<<k)]+=dp[i-1][j],我最开始用的滚动数组,果断3000多ms,然后一步一步优化降到300ms,最后发现可以用1维数组表示,类似背包思想。 本想继续优化的,但确实想不出什么好的方法了
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
using namespace std;
const int maxn=21;
const int maxm=1<<20;
int n,m,dp[maxm],s[maxn];
vector<int> g[maxn];
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
memset(dp,0,sizeof(dp));
memset(s,0,sizeof(s));
for(int i=1;i<=n;i++)
{
int num;
scanf("%d",&num);
g[i].clear();
while(num--)
{
int id;
scanf("%d",&id);
id--;
g[i].push_back(id);
s[i]|=1<<id;
}
}
int size=g[1].size();
for(int i=0;i<size;i++)
dp[1<<g[1][i]]=1;
int tol=1<<m;
for(int i=2;i<=n;i++)
{
int size=g[i].size();
for(int j=tol-1;j>=0;j--)
{
if(!dp[j])
continue;
if(((~j)&s[i])==0)
{
dp[j]=0;
continue;
}
for(int k=0;k<size;k++)
{
if((j>>g[i][k])&1)
continue;
dp[j|(1<<g[i][k])]+=dp[j];
}
dp[j]=0;
}
}
int ans=0;
for(int i=0;i<tol;i++)
ans+=dp[i];
printf("%d\n",ans);
}
return 0;
}