BZOJ1059 矩阵游戏

BZOJ1059 矩阵游戏

题目传送门

题解

刚开始理解错题意了,以为除了主对角线上之外都不能有1.后来发现只要主对角线上可以都是1就是合法的了。实际上的题意就是让你判断\(n\)行与\(n\)列是否可以都匹配上,这样就是一个比较简单的二分图匹配,如果第\((i,j)\)位置上为1,就说明第\(i\)行与第\(j\)行能够匹配,然后跑一遍匈牙利就行了。

code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
bool Finish_read;
template<class T>inline void read(T &x){Finish_read=0;x=0;int f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;if(ch==EOF)return;ch=getchar();}while(isdigit(ch))x=x*10+ch-'0',ch=getchar();x*=f;Finish_read=1;}
template<class T>inline void print(T x){if(x/10!=0)print(x/10);putchar(x%10+'0');}
template<class T>inline void writeln(T x){if(x<0)putchar('-');x=abs(x);print(x);putchar('\n');}
template<class T>inline void write(T x){if(x<0)putchar('-');x=abs(x);print(x);}
/*================Header Template==============*/
const int maxn=205;
int T,n,tot;
int a[maxn][maxn];
struct edge {
	int to,nxt;
}E[maxn*maxn];
int head[maxn],belong[maxn],vis[maxn];
/*==================Define Area================*/
void addedge(int u,int v) {
	E[++tot].nxt=head[u];head[u]=tot;E[tot].to=v;
}

void init() {
	memset(head,-1,sizeof head);
	memset(belong,0,sizeof belong);
	tot=0;
}

bool Solve(int u) {
	for(int i=head[u];~i;i=E[i].nxt) {
		int to=E[i].to;
		if(vis[to]) continue;
		vis[to]=1;
		if(!belong[to]||Solve(belong[to])) {
			return belong[to]=u;
		}
	}
	return 0;
}

int main() {
	read(T);
	while(T--) {
		read(n);
		init();
		for(int i=1;i<=n;i++) {
			for(int j=1;j<=n;j++) {
				read(a[i][j]);
				if(a[i][j]) addedge(i,j);
			}
		}
		int ans=0;
		for(int i=1;i<=n;i++) {
			memset(vis,0,sizeof vis);
			ans+=Solve(i);
		}
		if(ans==n) puts("Yes");
		else puts("No");
	}
	return 0;
}
posted @ 2018-08-06 19:51  Apocrypha  阅读(174)  评论(0编辑  收藏  举报