洛谷P1129 [ZJOI2007]矩阵游戏——题解

题目传送门
题目大意：
n*n的棋盘，每格为白色或黑色，问是否能通过任意交换两行、两列使得左上角到右下角的一条对角线上全为黑色。

---

思考过程：
我们如果要将一个格子移到满足条件的地方，相当于让它所在的行的一列被匹配到。而最终的情况就是每一行都有一列被匹配，并且被匹配的列不能重复。那么我们就很自然的想到了二分图匹配，以一个1格子为边，连接他所在的行与列，然后跑二分图匹配。

---

具体做法：
1.对每一个1格子，建一条边连接他所在的行与列
2.跑二分图匹配
3.如果匹配数恰好等于n，那么就是Yes，否则为No

---

代码：

#include <bits/stdc++.h>
using namespace std;

const int maxn=250;
struct stu
{
    int to,next;    
}road[maxn*maxn*2]; int first[maxn*2],cnt=0;
int map1[maxn][maxn],match[maxn*2],T,n;
bool book[2*maxn];

void init()
{
    cnt=0;
    memset(first,0,sizeof(first));
    memset(match,0,sizeof(match));
}

void addedge(int x,int y)
{
    road[++cnt].to=y;
    road[cnt].next=first[x];
    first[x]=cnt;   
}

bool dfs(int now)
{
    for(int i=first[now];i;i=road[i].next)
    {
        int to=road[i].to;
        if(!book[to])
        {
            book[to]=1;
            if(!match[to]||dfs(match[to]))
            {
                match[to]=now;
                match[now]=to;
                return 1;
            }
        }
    }
    return 0;
}

int main()
{
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                scanf("%d",&map1[i][j]);
                if(map1[i][j]==1) { addedge(i,j+n);addedge(j+n,i); }
            }
        }
        int cnt=0;
        for(int i=1;i<=n;i++)
        {
            if(!match[i])
            {
                memset(book,0,sizeof(book));
                book[i]=1;
                if(dfs(i)) cnt++;
            }
        }
        if(cnt==n) { printf("Yes\n");continue; }
        printf("No\n");
    }
    return 0;   
}