题目描述:
给出两列士兵 第一行的i可以与第二行的j ( |i-j|<=e ) 配对 其中有k对不能匹配 现在求有多少种完全匹配方案 对1e9+7取模
题目分析:
看到e<=4也就是说某一个人能跟他匹配的很少 那么就可以用状压来表示第i个第一行的数字可以与其匹配的数字的匹配状态 然后加个记忆化搜索就可以了
题目链接:
代码:
#include <cstdio>
#include <iostream>
#include <cstring>
#define int long long
const int mod=1e9+7;
const int maxm=2011;
int dp[maxm][1<<10],ms[maxm][maxm];
int n,k,e;
int dfs(int now,int sta)
{
if(now>n) return 1;
sta>>=1;
if(~dp[now][sta]) return dp[now][sta];
int ans=0;
for(int i=0;i<=2*e;i++)
if(now-e+i>=1&&now-e+i<=n)
{
if((1<<i)&sta) continue;
if(ms[now][now-e+i]) continue;
ans=(ans+dfs(now+1,(sta|(1<<i))))%mod;
}
dp[now][sta]=ans;
return ans;
}
signed main()
{
memset(dp,-1,sizeof(dp));
scanf("%lld%lld%lld",&n,&e,&k);
for(int i=1,x,y;i<=k;i++)
scanf("%lld%lld",&x,&y),ms[x][y]=1;
printf("%lld",dfs(1,0));
return 0;
}