[ZJOI2007]矩阵游戏

Posted 2021-02-21 cutemush

小Q是一个非常聪明的孩子，除了国际象棋，他还很喜欢玩一个电脑益智游戏——矩阵游戏。矩阵游戏在一个N

*N黑白方阵进行（如同国际象棋一般，只是颜色是随意的）。每次可以对该矩阵进行两种操作：行交换操作：选择
矩阵的任意两行，交换这两行（即交换对应格子的颜色）列交换操作：选择矩阵的任意行列，交换这两列（即交换
对应格子的颜色）游戏的目标，即通过若干次操作，使得方阵的主对角线(左上角到右下角的连线)上的格子均为黑
色。对于某些关卡，小Q百思不得其解，以致他开始怀疑这些关卡是不是根本就是无解的！！于是小Q决定写一个程
序来判断这些关卡是否有解。
Input
　　第一行包含一个整数T，表示数据的组数。接下来包含T组数据，每组数据第一行为一个整数N，表示方阵的大
小；接下来N行为一个N*N的01矩阵（0表示白色，1表示黑色）。
N ≤ 200
Output
　　输出文件应包含T行。对于每一组数据，如果该关卡有解，输出一行Yes；否则输出一行No。

Sample Input
2
2
0 0
0 1
3
0 0 1
0 1 0
1 0 0
Sample Output
No
Yes

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
int line[2000][2010];
int used[2010];
int bai[2020];
int tag;
int n;
int ans;
bool find(int x)
{
    for(int i=1;i<=n;i++)
    {
        if(line[x][i]&&used[i]!=tag)
        {
            used[i]=tag;
            if(bai[i]==0||find(bai[i]))
            {
                bai[i]=x;
                return true;
            }
        }
    }
    return false;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        ans=0;//here
        tag=0;
        memset(line,0,sizeof line);
        memset(bai,0,sizeof bai);
        memset(used,0,sizeof used);
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&line[i][j]);
        for(int i=1;i<=n;i++)
        {
            tag++;
            if(find(i))
                ans++;
        }
        if(ans==n) cout<<"Yes"<<endl;
        else cout <<"No"<<endl;
    }
    return 0;
}