【bzoj1059】 [ZJOI2007]矩阵游戏 匈牙利算法

题解

在同一行同一列的1无论怎么换来换去都会在同一行同一列，
如果想通过转换变到主对角线上必须满足有n个点不在同一行不在同一列，那么我们以i为一边以j为一边进行二分图匹配即可。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cctype>
using namespace std;
int mp[1010][1010],con[1010],visit[1010],T,n;
inline int read()
{
    int ret=0;char ch=getchar();
    for (;!isdigit(ch);ch=getchar());
    for (;isdigit(ch);ch=getchar()) ret=ret*10+ch-'0';
    return ret;
}
bool judge(int x)
{
    for (int i=1;i<=n;i++)
    {
        if ((!visit[i])&&mp[x][i])
        {
            visit[i]=1;
            if ((!con[i])||judge(con[i]))
            {
                con[i]=x;
                return true;
            }
        }
    }
    return false;
}
bool check()
{
    for (int i=1;i<=n;i++)
    {
        memset(visit,0,sizeof(visit));
        if (!judge(i)) return false;
    }
    return true;
}
int main()
{
    for (T=read();T;T--)
    {
        memset(mp,0,sizeof(mp));
        memset(con,0,sizeof(con));
        n=read();
        for (int i=1;i<=n;i++)
            for (int j=1;j<=n;j++)
            {
                int x=read();
                if (x) mp[i][j]=1;
            }
        if (check()) printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}