BZOJ1059 矩阵游戏

• BZOJ1059 矩阵游戏
• 题解
• ｃｏｄｅ

BZOJ1059 矩阵游戏

题目传送门

题解

刚开始理解错题意了，以为除了主对角线上之外都不能有１．后来发现只要主对角线上可以都是１就是合法的了。实际上的题意就是让你判断\(n\)行与\(n\)列是否可以都匹配上，这样就是一个比较简单的二分图匹配，如果第\((i,j)\)位置上为１，就说明第\(i\)行与第\(j\)行能够匹配，然后跑一遍匈牙利就行了。

ｃｏｄｅ

#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;
}