BZOJ 1059: [ZJOI2007]矩阵游戏 匈牙利算法

1059: [ZJOI2007]矩阵游戏

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 2351  Solved: 1156

题目连接

http://www.lydsy.com/JudgeOnline/problem.php?id=1059

Description

小Q是一个非常聪明的孩子，除了国际象棋，他还很喜欢玩一个电脑益智游戏——矩阵游 戏。矩阵游戏在一个N*N黑白方阵进行（如同国际象棋一般，只是颜色是随意的）。每次可以对该矩阵进行两种操作：行交换操作：选择矩阵的任意两行，交换这 两行（即交换对应格子的颜色）列交换操作：选择矩阵的任意行列，交换这两列（即交换对应格子的颜色）游戏的目标，即通过若干次操作，使得方阵的主对角线 (左上角到右下角的连线)上的格子均为黑色。对于某些关卡，小Q百思不得其解，以致他开始怀疑这些关卡是不是根本就是无解的！！于是小Q决定写一个程序来 判断这些关卡是否有解。

 

Input

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

Output

输出文件应包含T行。对于每一组数据，如果该关卡有解，输出一行Yes；否则输出一行No。

Sample Input

2
2
0 0
0 1
3
0 0 1
0 1 0
1 0 0

Sample Output

No
Yes
【数据规模】
对于100%的数据，N ≤ 200

HINT

 

题解：

建图，对于每一个坐标，如果g[i][j]=1的话，那么我们add_edge(i,j)

然后跑一发匈牙利算法，如果对于每一个坐标，如果每一行都能有一个找到对象的话，那么说就是可行的

~\(≧▽≦)/~啦啦啦

虽然这道题我莫名其妙的狂WA不止，不过思路还是很简单的= =

代码：

 

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
struct node{
     int next,to;
     }e[40010];
int match[410],p[410],T,i,j,n,tot;
bool visit[410];
void add(int u,int v)
{
   e[++tot].to=v;
   e[tot].next=p[u];
   p[u]=tot;
}
bool dfs(int u)
{
   for (int i=p[u]; i!=0; i=e[i].next)
   {
     int v=e[i].to;
     if (!visit[v])
     {
       visit[v]=true;
       if (match[v]==0 || dfs(match[v]))
       {
         match[v]=u;
         return true;
       }
     }
   }
   return false;
}
int main()
{
  scanf("%d",&T);
  while (T--)
  {
      tot=0;
      memset(match,0,sizeof(match));
      memset(p,0,sizeof(p));
      scanf("%d",&n);
      for (i=1; i<=n; i++)
        for (j=1; j<=n; j++)
        {
          int c;
          scanf("%d",&c);
          if (c==1) add(i,j+n);
        }
      for (i=1; i<=n; i++)
      {
        memset(visit,false,sizeof(visit));
        if (!dfs(i))  break;
      }
      if (i>n)  printf("Yes\n"); else printf("No\n");
  }
}