Description
N 个男孩,N 个女孩,男孩和女孩可能是朋友,也可能不是朋友。现在要组成N 对舞伴,要求每对舞
伴都是一男一女,且他们是朋友。
统计不同配对方案的数量,因为结果很大,所以只要求除以M 的余数。
Input
第1 行,2 个整数N,M。接下来N 行,每行N 个整数Aij,表示第i 个男孩和第j 个女孩的关系。如果他们是朋友,则Aij = 1,否则Aij = 0。
Output
1 个整数,表示所求的值。
Sample Input
3 1000000000
1 1 1
1 1 1
1 1 1
Sample Output
6
Data Constraint
• 对于50% 的数据,N <= 9;
• 对于100% 的数据,1 <= N <= 20, 1 <= M <= 10^9; 0 <= Aij <= 1。
。。一开始看题意,以为这是二分图完美匹配的方案数,woc,没见过不会啊!
后来一看n<=20。。感觉被骗了。。直接傻逼状态dp。。设f[i][s]表示前i个boy的方案数,s存储的是girl的状态。然后跑了一发大数据发现被卡了。。因为第一遍枚举boy,第二遍枚举边,第三遍枚举状态,时间复杂度就是20*2^n*20,达到了4e8。。肯定跑不过。
那么我们就压位,把边也压缩成二进制,这种方法在高级的状压dp中经常使用。。设boy[i]表示矩阵第i行的状态,其实含义上就代表第i个男孩跟所有女孩的状态,那么记忆化搜索即可,因为直接dp复杂度会退化,有很多无用状态。。
#include<cstdio>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=21;
bool v[22][1<<N];
int f[22][1<<N];
int boy[22],n,m;
typedef long long ll;
int dfs(int x,int now)
{
if (x==n)return 1;
if (v[x][now])return f[x][now];
v[x][now]=1;
int s=now&boy[x],ans=0;
while (s)
{
int edge=s&(-s);
ans=(ans+dfs(x+1,now-edge))%m;
s-=edge;
}
return f[x][now]=ans;
}
int main()
{
freopen("perm.in", "r", stdin);
freopen("perm.out", "w", stdout);
scanf("%d%d",&n,&m);
fo(i,0,n-1)
fo(j,0,n-1)
{
int x;
scanf("%d",&x);
if (x)boy[i]+=1<<j;
}
ll ans=dfs(0,(1<<n)-1);
printf("%lld\n",ans);
return 0;
}