BZOJ1059 矩阵游戏
题解
刚开始理解错题意了,以为除了主对角线上之外都不能有1.后来发现只要主对角线上可以都是1就是合法的了。实际上的题意就是让你判断\(n\)行与\(n\)列是否可以都匹配上,这样就是一个比较简单的二分图匹配,如果第\((i,j)\)位置上为1,就说明第\(i\)行与第\(j\)行能够匹配,然后跑一遍匈牙利就行了。
code
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
bool Finish_read;
template<class T>inline void read(T &x){Finish_read=0;x=0;int f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;if(ch==EOF)return;ch=getchar();}while(isdigit(ch))x=x*10+ch-'0',ch=getchar();x*=f;Finish_read=1;}
template<class T>inline void print(T x){if(x/10!=0)print(x/10);putchar(x%10+'0');}
template<class T>inline void writeln(T x){if(x<0)putchar('-');x=abs(x);print(x);putchar('\n');}
template<class T>inline void write(T x){if(x<0)putchar('-');x=abs(x);print(x);}
/*================Header Template==============*/
const int maxn=205;
int T,n,tot;
int a[maxn][maxn];
struct edge {
int to,nxt;
}E[maxn*maxn];
int head[maxn],belong[maxn],vis[maxn];
/*==================Define Area================*/
void addedge(int u,int v) {
E[++tot].nxt=head[u];head[u]=tot;E[tot].to=v;
}
void init() {
memset(head,-1,sizeof head);
memset(belong,0,sizeof belong);
tot=0;
}
bool Solve(int u) {
for(int i=head[u];~i;i=E[i].nxt) {
int to=E[i].to;
if(vis[to]) continue;
vis[to]=1;
if(!belong[to]||Solve(belong[to])) {
return belong[to]=u;
}
}
return 0;
}
int main() {
read(T);
while(T--) {
read(n);
init();
for(int i=1;i<=n;i++) {
for(int j=1;j<=n;j++) {
read(a[i][j]);
if(a[i][j]) addedge(i,j);
}
}
int ans=0;
for(int i=1;i<=n;i++) {
memset(vis,0,sizeof vis);
ans+=Solve(i);
}
if(ans==n) puts("Yes");
else puts("No");
}
return 0;
}