P1129 [ZJOI2007]矩阵游戏
题目描述
小\(Q\)是一个非常聪明的孩子,除了国际象棋,他还很喜欢玩一个电脑益智游戏――矩阵游戏。矩阵游戏在一个\(N*N\)黑白方阵进行(如同国际象棋一般,只是颜色是随意的)。每次可以对该矩阵进行两种操作:
行交换操作:选择矩阵的任意两行,交换这两行(即交换对应格子的颜色)
列交换操作:选择矩阵的任意两列,交换这两列(即交换对应格子的颜色)
游戏的目标,即通过若干次操作,使得方阵的主对角线(左上角到右下角的连线)上的格子均为黑色。
对于某些关卡,小\(Q\)百思不得其解,以致他开始怀疑这些关卡是不是根本就是无解的!!于是小\(Q\)决定写一个程序来判断这些关卡是否有解。
输入输出格式
输入格式:
第一行包含一个整数\(T\),表示数据的组数。
接下来包含\(T\)组数据,每组数据第一行为一个整数\(N\),表示方阵的大小;接下来\(N\)行为一个\(N*N\)的\(01\)矩阵(\(0\)表示白色,\(1\)表示黑色)。
输出格式:
包含\(T\)行。对于每一组数据,如果该关卡有解,输出一行\(Yes\);否则输出一行\(No\)。
说明
对于\(20\)%的数据,\(N ≤ 7\)
对于\(50\)%的数据,\(N ≤ 50\)
对于\(100\)%的数据,\(N ≤ 200\)
研究一下操作和要求,我们可以这么转化:每行至少有一个\(1\)并且这些\(1\)互相不在同一列
对于行\(i\),我们用了坐标\((i,j)\)的\(1\),那么第\(j\)列的\(1\)不就废了吗?
好的,求匹配。
坐标为\((i,j)\)的点作为第\(i\)行第\(j\)列的边。
求每一行和每一列的最大匹配数即可。
二分图匹配。
code:
#include <cstdio>
#include <cstring>
const int N=202;
struct Edge
{
int to,next;
}g[N*N];
int T,head[N],cnt=0,n;
void add(int u,int v)
{
g[++cnt].to=v,g[cnt].next=head[u],head[u]=cnt;
}
int match[N],used[N];
bool m_match(int now)
{
for(int i=head[now];i;i=g[i].next)
{
int v=g[i].to;
if(!used[v])
{
used[v]=1;
if(!match[v]||m_match(match[v]))
{
match[v]=now;
return true;
}
}
}
return false;
}
int main()
{
scanf("%d",&T);
int is;
while(T--)
{
memset(match,0,sizeof(match));
memset(head,0,sizeof(head));
scanf("%d",&n);cnt=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
scanf("%d",&is);
if(is) add(i,j);
}
int ans=0;
for(int i=1;i<=n;i++)
{
memset(used,0,sizeof(used));
if(m_match(i)) ans++;
}
if(ans==n) printf("Yes\n");
else printf("No\n");
}
}
我的一些出现的细节错误:
- 匈牙利used数组置true时机(似乎经常错)
- used置0的时机(在if的外面)
- 前向星head,cnt置0
2018.5.5