正题
题目链接:https://www.luogu.com.cn/problem/P1129
题目大意
n∗mn*mn∗m的网格上有0/10/10/1,可以选择交换两行或两列,求能否让对角线上全是111。
解题思路
因为可以交换,所以是每一行都要和每一列配对。而第iii行能够和第jjj列配对的条件是(i,j)(i,j)(i,j)这个格子上有111,然后跑网络流最大匹配即可。
codecodecode
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int N=210*210,inf=2147483647/3;
struct node{int to,next,w;
}a[N*2];
int T,n,s,t,tot,dep[N],ls[N],ans;
queue<int> q;
void addl(int x,int y,int w){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;a[tot].w=w;a[++tot].to=x;a[tot].next=ls[y];ls[y]=tot;a[tot].w=0;return;
}
bool bfs(){memset(dep,0,sizeof(dep));dep[s]=1;while(!q.empty())q.pop();q.push(s);while(!q.empty()){int x=q.front();q.pop();for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(dep[y]||!a[i].w)continue;dep[y]=dep[x]+1;if(y==t)return 1;q.push(y);}}return 0;
}
int dinic(int x,int flow){int rest=0,k;if(x==t)return flow;for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(dep[x]+1!=dep[y]||!a[i].w)continue;rest+=(k=dinic(y,min(flow-rest,a[i].w)));a[i].w-=k;a[i^1].w+=k;if(rest==flow)return flow;}if(!rest)dep[x]=0;return rest;
}
int main()
{scanf("%d",&T);while(T--){tot=1;ans=0;memset(ls,0,sizeof(ls));scanf("%d",&n);s=2*n+1;t=s+1;for(int i=1;i<=n;i++)for(int j=1;j<=n;j++){int w;scanf("%d",&w);if(w)addl(i,j+n,1);}for(int i=1;i<=n;i++)addl(s,i,1),addl(i+n,t,1);while(bfs())ans+=dinic(s,inf);if(ans==n)printf("Yes\n");else printf("No\n");}return 0;
}