[ZJOI2007]矩阵游戏

题目

https://www.luogu.org/problemnew/show/P1129

思路

我们发现,我们把点看成匹配边的话,题目是要求每行每列要相互配对。

若a[i][j]=1则add(i,j+n)

代码

#include
#include
#include
using namespace std;
const int maxn=377;
int list[maxn*2],t,n,ans,cnt,match[maxn*2];
bool b[maxn*2];
struct E
{
	int to,next;
}e[maxn*maxn*3];
void add(int u,int v)
{
	e[++cnt].to=v; e[cnt].next=list[u]; list[u]=cnt;
}
bool dfs(int u)
{
	for(int i=list[u]; i; i=e[i].next)
	{
		int v=e[i].to;
		if(!b[v])
		{
			b[v]=1;
			if(!match[v]||dfs(match[v]))
			{
				match[v]=u; return 1;
			}
		}
	}
	return 0;
}
int main()
{
	scanf("%d",&t);
	while(t--)
	{
		memset(list,0,sizeof(list)); memset(e,0,sizeof(e)); memset(match,0,sizeof(match)); ans=0; cnt=0;
		scanf("%d",&n);
		for(int i=1; i<=n; i++) for(int j=1; j<=n; j++)
		{
			int x;
			scanf("%d",&x);
			if(x) add(i,j+n),add(j+n,i);
		}
		for(int i=1; i<=n; i++)
		{
			memset(b,0,sizeof(b));
			if(dfs(i)) ans++;
		}
		if(ans>=n) printf("Yes\n");else printf("No\n");
	}
}

你可能感兴趣的:(题解)