【匈牙利算法】BZOJ1059-[ZJOI2007]矩阵游戏

【题目大意】

给出一个局部染色的矩阵，问能否通过交换行或者列使得最后又一条对角线全部被染色过？

【思路】

无论如何交换，同一行的格子依然在同一行，同一列的格子依然在同一列。所以只需找出n个行号列号均不同的格子即可，裸的二分图匹配。

【错误点】

初始化出了问题。p忘记初始化+开始我将lk的初始值设为0，但是下标又是从0开始的，所以会出现lk[0]=0,其实它已经匹配过了，会出错所以将初始值改为了-1。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=205;
int p[MAXN][MAXN];
int vis[MAXN],lk[MAXN];
int n;

int find(int u)
{
    for (int i=0;i<n;i++)
        if (!vis[i] && p[u][i])
        {
            vis[i]=1;
            if (lk[i]==-1 || find(lk[i]))
            /*一开始我将lk的初始值设为0，但是下标又是从0开始的，所以会出现lk[0]=0,其实它已经匹配过了，会出错
              所以将初始值改为了-1*/
            {
                lk[i]=u;
                return 1;
            }
        }
    return 0;
}

void Hungary()
{
    memset(lk,-1,sizeof(lk));
    int ans=0;
    for (int i=0;i<n;i++)
    {
        memset(vis,0,sizeof(vis));
        if (find(i)) ans++;else break;
    }
    if (ans==n) cout<<"Yes"<<endl;else cout<<"No"<<endl;
}

void init()
{
    memset(p,0,sizeof(p));
    /*这里忘记初始化了，多组数据的初始化要注意*/ 
    scanf("%d",&n);
    for (int i=0;i<n;i++)
        for (int j=0;j<n;j++)
        {
            int x;
            scanf("%d",&x);
            if (x==1) 
            {
                p[i][j]=1;
            }
        }
}

int main()
{
    int T;
    scanf("%d",&T);
    for (int kase=0;kase<T;kase++)
    {
        init();
        Hungary();
    }
    return 0;
}