洛谷 P1129 [ZJOI2007]矩阵游戏 二分图 匈牙利算法 链式前向星 读入优化

题目链接：
https://www.luogu.com.cn/problem/P1129

参考博客：

https://www.luogu.com.cn/blog/sswcdak/solution-p1129

这个博客有图形解释很清晰，可以复习再看

算法：1：匈牙利算法模板 2：链式前向星

思路：

1：思路就一句话,每一行（男孩子）匹配每一列（妹子），等于老师讲的每一个男孩子分配一个妹子

#include <bits/stdc++.h>
//思路就一句话,每一行匹配每一列，等于老师讲的每一个男孩子分配一个妹子
//相当于有2*n个点即最多为400个
using namespace std;
const int maxn=4e2+1;
int head[maxn],to[maxn*maxn],next[maxn*maxn],tot=1,ans=0,match[maxn],visit[maxn],n,t;
//maxn==401,为什么不等于201，因为match数组和visit数组[]内需要写到400
//1到200表示的是男孩子，201到400表示的是妹子
//一定要熟记男孩子找妹子那个模型

inline int read()//读入优化
{
    int s=0,w=1;
    char g=getchar();
    while(g<'0'||g>'9')
    {
        if(g=='-')w*=-1;
        g=getchar();
    }
    while(g>='0'&&g<='9')
    {
        s=s*10+g-'0';
        g=getchar();
    }
    return s*w;
}

void add(int x,int y)//链式前向星
{
    tot++;
    to[tot]=y;
    next[tot]=head[x];
    head[x]=tot;
}

bool dfs(int x)//匈牙利算法给每一个男孩子匹配妹子
{
    for(int i=head[x],y;i;i=next[i])
    {
        if(!visit[y=to[i]])
        {
            visit[y]=1;
            if(!match[y]||dfs(match[y]))
            {
                match[y]=x;
                return true;
            }
        }
    }
    return false;
}

void clear()//多组数据，清零初始化
{
    t--;
    ans=0;
    for(int i=1;i<=tot;i++)to[i]=head[i]=next[i]=0;//这种方式可以避免memset()浪费过多时间
    tot=1;
    for(int i=n+1;i<=2*n;i++)match[i]=0;//match[i]数组只存妹子被匹配的男孩子,i的范围是妹子节点的编号n+1到2*n
}

int main()
{
    ios::sync_with_stdio(0);
    t=read();
    while(t!=0)
    {
        n=read();
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
        {
            int m1=read();//读入并建边
            if(m1==1)add(i,j+n);
        }
        for(int i=1;i<=n;i++)
        {
            memset(visit,0,sizeof(visit));
            if(dfs(i))ans++;
        }
        if(ans==n)printf("Yes\n");//有n对匹配
        else printf("No\n");
        clear();
    }
    return 0;
}