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");

  }

}