再次根据数据猜算法,233,网络流跑二分图匹配吧。
每个人需要一张床,所以就是床与人匹配。
然后根据题目描述来建边。
#include<bits/stdc++.h>
#define N 150
#define M 3000
#define inf ~0u>>1
using namespace std;
int n,cnt,T,S,bed[N+1],x,ans,Case;
int first[N+1],nex[M+1],to[M+1],fl[M+1],siz,dis[N+1];
inline char nc()
{
static char buf[100000],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
int x=0,b=1;
char c=nc();
for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
return x*b;
}
inline void add(int x,int y,int z)
{
nex[siz]=first[x];
first[x]=siz;
to[siz]=y;
fl[siz]=z;
siz++;
}
inline void insert(int x,int y,int z)
{
add(x,y,z),add(y,x,0);
}
inline bool BFS(int s,int t)
{
queue <int> Q;
for(int i=S;i<=T;i++)dis[i]=inf;
dis[s]=0,Q.push(s);
while(!Q.empty())
{
int x=Q.front();Q.pop();
for(int i=first[x];i!=-1;i=nex[i])
{
int u=to[i];
if(fl[i]&&dis[x]+1<dis[u])
{
dis[u]=dis[x]+1;
if(u==t)return true;
Q.push(u);
}
}
}
if(dis[t]==inf)return false;
return true;
}
inline int DFS(int s,int f,int t)
{
if(s==t)return f;
int rest=f;
for(int i=first[s];i!=-1;i=nex[i])
{
int u=to[i];
if(fl[i]&&rest&&dis[u]==dis[s]+1)
{
int tmp=DFS(u,min(fl[i],rest),t);
if(tmp==0)dis[u]=0;
fl[i]-=tmp;
fl[i^1]+=tmp;
rest-=tmp;
}
}
return f-rest;
}
inline int Dinic(int s,int t)
{
int ans=0;
while(BFS(s,t))ans+=DFS(s,inf,t);
return ans;
}
int main()
{
//freopen("in.txt","r",stdin);
Case=read();
while(Case--)
{
cnt=0,siz=0;
memset(first,-1,sizeof(first));
n=read();
S=0,T=2*n+1;
for(int i=1;i<=n;i++)bed[i]=read();
for(int i=1;i<=n;i++)if(bed[i])insert(i+n,T,1);
for(int i=1;i<=n;i++)
{
x=read();
if((!x&&bed[i])||!bed[i])insert(S,i,1),cnt++;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
x=read();
if(x||i==j)insert(i,j+n,1);
}
ans=Dinic(S,T);
if(ans==cnt)puts("^_^");
else puts("T_T");
}
return 0;
}
36ms,然而他们28ms如何写出来的啊233。