二分图匹配匈牙利算法(poj1469验证)

读入的时候就进行匹配,在这个匹配的基础上寻找增广路,更新之

#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;
const int MAX=400;
int cx[MAX],cy[MAX];
int sx[MAX],sy[MAX];
int g[MAX][MAX];
int t,n,m;
int dfs(int k){
    cx[k]=1;
    for(int i=1;i<=m;++i){
        if(g[k][i]!=-1&&cy[i]==-1){
            cy[i]=1;
            if(sy[i]==-1||dfs(sy[i])){
                sy[i]=k;
                sx[k]=i;
                return 1;
            }
        }
    }
    return 0;
}

int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        int ans=0;
        memset(sx,-1,sizeof(sx));
        memset(sy,-1,sizeof(sy));
        memset(g,-1,sizeof(g));
        for(int i=1;i<=n;++i){
            int k; scanf("%d",&k);
            while(k--){
                int j;
                scanf("%d",&j);
                g[i][j]=1;
                if(sx[i]==-1&&sy[j]==-1){
                    sx[i]=j;
                    sy[j]=i;
                    ++ans;
                }
            }
        }
        for(int i=1;i<=n;++i){
            if(sx[i]==-1){
                memset(cx,-1,sizeof(cx));
                memset(cy,-1,sizeof(cy));
                ans+=dfs(i);
            }
        }
        //printf("%d\n",ans);
        if(ans==n) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}


你可能感兴趣的:(算法)