题目描述
小Q是一个非常聪明的孩子,除了国际象棋,他还很喜欢玩一个电脑益智游戏――矩阵游戏。矩阵游戏在一个N*N黑白方阵进行(如同国际象棋一般,只是颜色是随意的)。每次可以对该矩阵进行两种操作:
行交换操作:选择矩阵的任意两行,交换这两行(即交换对应格子的颜色)
列交换操作:选择矩阵的任意两列,交换这两列(即交换对应格子的颜色)
游戏的目标,即通过若干次操作,使得方阵的主对角线(左上角到右下角的连线)上的格子均为黑色。
对于某些关卡,小Q百思不得其解,以致他开始怀疑这些关卡是不是根本就是无解的!!于是小Q决定写一个程序来判断这些关卡是否有解。
输入输出格式
输入格式:
第一行包含一个整数T,表示数据的组数。
接下来包含T组数据,每组数据第一行为一个整数N,表示方阵的大小;接下来N行为一个N*N的01矩阵(0表示白色,1表示黑色)。
输出格式:
包含T行。对于每一组数据,如果该关卡有解,输出一行Yes;否则输出一行No。
输入输出样例
输入样例#1:
2 2 0 0 0 1 3 0 0 1 0 1 0 1 0 0
输出样例#1:
No Yes
说明
对于20%的数据,N ≤ 7
对于50%的数据,N ≤ 50
对于100%的数据,N ≤ 200
【题目分析】
匈牙利算法。如果line[i][j]==1可以理解为从i到j有边,最后看所有的i是否都能匹配完。
因为是多组数据,一定要记得初始化!!!初始化!!!(卡了半个小时,不开心)
#include <cstdio> #include <cstring> #include <iostream> using namespace std; int line[2000][2010]; bool used[2010]; int bai[2020]; int n; int ans; bool find(int x) { for(int i=1;i<=n;i++) { if(line[x][i]&&used[i]==0) { used[i]=1; 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 memset(line,0,sizeof line); memset(bai,0,sizeof bai); 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++) { memset(used,0,sizeof used); if(find(i)) ans++; } if(ans==n) cout<<"Yes"<<endl; else cout <<"No"<<endl; } return 0; } /* 2 2 0 0 0 1 3 0 0 1 0 1 0 1 0 0 */