BZOJ 1059: [ZJOI2007]矩阵游戏

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

分析

问题可以转换为有多少个1不再同一行同一列，于是我们把为1的格子的i向j连一条边再跑最大匹配即可。

PS：因为貌似时间不够了，于是就把以前网络流的板子c了下来。(^__^) 嘻嘻

代码

#include <bits/stdc++.h>

#define N 505
#define inf 0x7fffffff

struct NOTE
{
    int to,next,c;
}e[N*10000];

int cnt=1,n,ans,s,t;
int dis[N],last[N],cur[N];

void add(int u,int v,int c)
{
    e[++cnt].to=v;e[cnt].c=c;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].c=0;e[cnt].next=last[v];last[v]=cnt;
}

bool bfs()
{
    memset(dis,0,sizeof(dis));
    dis[s]=1;
    std::queue<int> q;
    q.push(s);
    while (!q.empty())
    {
        int u=q.front();
        q.pop();
        for (int i=last[u];i;i=e[i].next)
            if (e[i].c&&!dis[e[i].to])
            {
                dis[e[i].to]=dis[u]+1;
                if (e[i].to==t) return 1;
                q.push(e[i].to);
            }
    }
    return 0;
}

int dfs(int x,int maxf)
{
    if (x==t||!maxf) return maxf;
    int ret=0;
    for (int &i=cur[x];i;i=e[i].next)
        if (e[i].c&&dis[e[i].to]==dis[x]+1)
        {
            int f=dfs(e[i].to,std::min(maxf-ret,e[i].c));
            e[i].c-=f;
            e[i^1].c+=f;
            ret+=f;
            if (ret==maxf) break;
        }
    return ret;
}

void dinic()
{
    while (bfs())
    {
        for (int i=s;i<=t;i++)
            cur[i]=last[i];
        ans+=dfs(s,inf);
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        s=0;t=n*2+1;cnt=1;
        memset(last,0,sizeof(last));
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            {
                int x;
                scanf("%d",&x);
                if(x==1)
                    add(i,j+n,1);
            }
        for(int i=1;i<=n;i++)
        {
            add(s,i,1);
            add(i+n,t,1);
        }   

        ans=0;
        dinic();
        if(ans==n)
            printf("Yes\n");
            else printf("No\n");
    }
}