洛谷 2055 [ZJOI2009]假期的宿舍——二分图匹配

时间:2023-03-09 18:31:00
洛谷 2055 [ZJOI2009]假期的宿舍——二分图匹配

题目:https://www.luogu.org/problemnew/show/P2055

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=;
int T,n,hd[N],xnt,to[N*N],nxt[N*N],pre[N];
bool vis[N],hj[N],flag;
int rdn()
{
int ret=;char ch=getchar();
while(ch>''||ch<'')ch=getchar();
while(ch>=''&&ch<='')(ret*=)+=ch-'',ch=getchar();
return ret;
}
void add(int x,int y)
{
to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;
}
bool dfs(int cr)
{
for(int i=hd[cr],v;i;i=nxt[i])
if(!vis[v=to[i]])
{
vis[v]=;
if(!pre[v]||dfs(pre[v]))
{pre[v]=cr;return true;}
}
return false;
}
int main()
{
T=rdn();
while(T--)
{
memset(hd,,sizeof hd);xnt=;
memset(pre,,sizeof pre);flag=;
n=rdn();bool x;
for(int i=;i<=n;i++)
{
x=rdn();if(x)vis[i]=;else vis[i]=;
}
for(int i=;i<=n;i++)
{
x=rdn();if(!vis[i]){hj[i]=;continue;}
if(x)hj[i]=;
else hj[i]=,add(i,i);
}
for(int i=;i<=n;i++)
for(int j=;j<=n;j++)
{
x=rdn();if(x&&!hj[i]&&vis[j])add(i,j);
}
for(int i=;i<=n;i++) if(!hj[i])
{
memset(vis,,sizeof vis);
if(!dfs(i)){flag=;break;}
}
if(flag)printf("%c%c%c\n",,,);else printf("%c%c%c\n",,,);
}
return ;
}