P1129 [ZJOI2007]矩阵游戏（二分图最大匹配）

最新推荐文章于 2022-01-20 14:52:32 发布

ESTHERWXY

最新推荐文章于 2022-01-20 14:52:32 发布

阅读量277

点赞数

分类专栏：匈牙利算法文章标签：匈牙利算法

本文链接：https://blog.csdn.net/ESTHERWXY/article/details/102367385

版权

匈牙利算法专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目描述

小Q是一个非常聪明的孩子，除了国际象棋，他还很喜欢玩一个电脑益智游戏――矩阵游戏。矩阵游戏在一个N×N黑白方阵进行（如同国际象棋一般，只是颜色是随意的）。每次可以对该矩阵进行两种操作：

行交换操作：选择矩阵的任意两行，交换这两行（即交换对应格子的颜色）

列交换操作：选择矩阵的任意两列，交换这两列（即交换对应格子的颜色）

游戏的目标，即通过若干次操作，使得方阵的主对角线(左上角到右下角的连线)上的格子均为黑色。

对于某些关卡，小Q百思不得其解，以致他开始怀疑这些关卡是不是根本就是无解的！于是小Q决定写一个程序来判断这些关卡是否有解。

输入格式

第一行包含一个整数T，表示数据的组数。

接下来包含T组数据，每组数据第一行为一个整数N，表示方阵的大小；接下来N行为一个N×N的01矩阵（0表示白色，1表示黑色）。

输出格式

包含T行。对于每一组数据，如果该关卡有解，输出一行YesYesYes；否则输出一行NoNoNo。

输入输出样例

输入 #1

输出 #1

No
Yes

说明/提示

对于20%的数据，N≤7

对于50%的数据，N≤50

对于100%的数据，N≤200

题解

像这种行与列之间的匹配问题常常可以用二分图最大匹配来做

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=505;
int head[N],nxt[N],tot,adj[N];
int vis[N],match[N];
void add(int u,int v)
{
	tot++;
	nxt[tot]=head[u];
	head[u]=tot;
	adj[tot]=v;
}
bool find(int u)
{
	int i;
	for(i=head[u];i;i=nxt[i])
	{
		int v=adj[i];
		if(!vis[v])
		{
			vis[v]=1;
			if(match[v]==0||find(match[v]))
			{
				match[v]=u;
				return true;
			}
		}
	}
	return false;
}
int main()
{
	int t,n;
	scanf("%d",&t);
	while(t--)
	{
		memset(adj,0,sizeof(adj));
		memset(vis,0,sizeof(vis));
		memset(vis,0,sizeof(nxt));
		memset(head,0,sizeof(head));
		memset(match,0,sizeof(match));
		tot=0;
		scanf("%d",&n);
		int i,j,x;
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
			{
				scanf("%d",&x);
				if(x==1)
					add(i,j);
			}
		int ans=0;
		for(i=1;i<=n;i++)
		{
			memset(vis,0,sizeof(vis));
			if(find(i))
				ans++;
		}
		if(ans>=n)
			printf("Yes\n");
		else
			printf("No\n");
	}
	return 0;
}

数组一定不要看开小了啊啊啊