题意:n后面第一行数据为0的是外校学生,为1的是本校学生,第二行为0则且第一行为1的为本校留校学生,第二行为1切第一行为1的为本校离校学生,每个学生可以睡自己(当学生为本校学生)和认识的人的床,问能否做到每个人(外校学生+本校留校学生)都有床睡。再下面的n * n矩阵为学生i和学生j的关系,1为朋友,0为陌生人。
分析:这是一个最大流问题,我们可以利用最大流算法计算出有多少张床能被人睡。
思路:
以2n+1为源点,2n+2为汇点,建图:
先以每张床为起点,汇点为终点建边,费用为1;
再以源点为起点,每个需要床睡觉的在校人员为终点建边,费用为1;
最后遍历n*n矩阵,若满足则建一条由
到
且费用为1的边。
统计在校学生数量,再对建好的图跑一遍最大流Dinic,若两者相等则所有不回家的本校学生和来看他们的其他人都有地方住,输出"^_^",反之则输出"T_T".
注意:此方法数组至少要开n*2+3.
AC代码:
/*
author:wuzx
*/
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<utility>
#include<string>
#include<queue>
#include<map>
#include<set>
#include<iterator>
#include<iomanip>
#include<stack>
#include<cstdio>
#define ll long long
#define endl "\n"
using namespace std;
const int maxn=110;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
using namespace std;
int ren[maxn],xx[maxn];
int xs[maxn][maxn];
struct Edge{
int from,to,cap,flow;
};
struct Dinic{
int n,m,s,t;
vector<Edge> edge;
vector<int> g[maxn];
bool vis[maxn];
int d[maxn];
int cur[maxn];
void init(int n)
{
memset(d,0,sizeof d);
for(int i=1;i<maxn;i++)
g[i].clear();
edge.clear();
}
void ClearFlow()
{
for(int i=0;i<edge.size();i++)
edge[i].flow=0;
}
void add(int from,int to,int cap)
{
edge.push_back(Edge{from,to,cap,0});
edge.push_back(Edge{to,from,0,0});
m=edge.size();
g[from].push_back(m-2);
g[to].push_back(m-1);
}
bool bfs()
{
memset(vis,0,sizeof vis);
queue<int> q;
q.push(s);
vis[s]=1;
d[s]=0;
while(!q.empty())
{
int x=q.front();
q.pop();
for(int i=0;i<g[x].size();i++)
{
Edge& e=edge[g[x][i]];
if(!vis[e.to]&&e.cap>e.flow)
{
vis[e.to]=1;
d[e.to]=d[x]+1;
q.push(e.to);
}
}
}
return vis[t];
}
int dfs(int x,int a)
{
if(x==t||a==0)
return a;
int flow=0,f;
for(int& i=cur[x];i<g[x].size();i++)
{
Edge& e=edge[g[x][i]];
if(d[x]+1==d[e.to]&&(f=dfs(e.to,min(a,e.cap-e.flow)))>0)
{
e.flow+=f;
edge[g[x][i]^1].flow-=f;
flow+=f;
a-=f;
if(a==0)
break;
}
}
return flow;
}
int maxflow(int s,int t)
{
this->s=s;this->t=t;
int flow=0;
while(bfs())
{
memset(cur,0,sizeof cur);
flow+=dfs(s,inf);
}
return flow;
}
};
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
int t; cin>>t;
while(t--)
{
int n;
cin>>n;
Dinic slove;
slove.init(n);
int s=2*n+1,t=2*n+2;
int ans=0;
for(int i=1;i<=n;i++)
{
cin>>ren[i];
if(ren[i]==1)
slove.add(i+n,t,1);
else
{
ans++;
slove.add(s,i,1);
}
}
for(int i=1;i<=n;i++)
{
cin>>xx[i];
if(xx[i]==0&&ren[i]==1)
{
ans++;
slove.add(s,i,1);
}
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
cin>>xs[i][j];
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
if(xs[i][j]==1||i==j)
slove.add(i,j+n,1);
}
int l=slove.maxflow(s,t);
cout<<(l==ans?"^_^":"T_T")<<endl;
}
return 0;
}