BZOJ 1059, 矩阵游戏

Problem

传送门

Mean

给定一个大小为N×N的黑白染色矩阵，试判断能否通过交换行与列使矩阵左上角至右下角这一对角线上所有点均为黑色。

Analysis

易知若几个黑点原本同行或同列，不管如何变换，它们始终同行或同列。因此我们只需判断是否存在N个点互不同行，也互不同列。
由此转化为二分图模型。对于一个坐标为(i,j)的黑点，由第i行向第j列连边。若此二分图存在完美匹配，则可满足题目要求。
跑网络流或者使用匈牙利算法均可。

Code

#include<cstdio>
const int N=205;
int t,n,x,ed,ans,v[N*N],nxt[N*N],g[N],f[N];
bool b[N];
void read(int &x){
    char c;
    while((c=getchar())<'0' || c>'9');
    x=c-'0';
    while((c=getchar())>='0' && c<='9') x=x*10+c-'0';
}
void add(int x,int y){
    v[++ed]=y;
    nxt[ed]=g[x];
    g[x]=ed;
}
bool find(int x){
    for(int i=g[x];i;i=nxt[i]) if(!b[v[i]]){
        b[v[i]]=1;
        if(!f[v[i]] || find(f[v[i]])) return f[v[i]]=x,1;
    }
    return 0;
}
int main(){
    read(t);
    while(t--){
        ed=ans=0;
        for(int i=1;i<=n;i++) g[i]=0;
        read(n);
        for(int i=1;i<=n;i++) for(int j=1;j<=n;j++){
            read(x);
            if(x) add(i,j);
        }
        for(int i=1;i<=n;i++) f[i]=0;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++) b[j]=0;
            if(find(i)) ans++;
        }
        printf("%s\n",ans==n?"Yes":"No");
    }
    return 0;
}