poj 1719 简单的匹配

一道简单的匹配问题，行列匹配，根据题意若行大于列匹配不成功，若列大于行，大于行的部分可随意。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int map[1001][2],col[1001],match[1001],visit[1001];
int dfs(int i)
{
    int j,k;
    for(j=0;j<2;j++)
    {
        k=map[i][j];
        if(!visit[k])
        {
            visit[k]=1;
            if(match[k]==0 || dfs(match[k]))
            {
                match[k]=i;
                col[i]=k;
                return 1;
            }
        }
    }
    return 0;
}
int main()
{
    int i,j,k,m,n,Case;
    scanf("%d",&Case);
    while(Case--)
    {
        scanf("%d %d",&n,&m);
        memset(match,0,sizeof(match));
        for(i=1;i<=m;i++)
        {
            scanf("%d %d",&map[i][0],&map[i][1]);
        }
        if(n>m)
        {
            printf("NO\n");
            continue;
        }
        for(i=1;i<=n;i++)
        {
            memset(visit,0,sizeof(visit));
            if(dfs(i)==0) break;
        }
        if(i==n+1)
        {
            printf("%d",col[1]);
            for(i=2;i<=n;i++)
                printf(" %d",col[i]);
            for(i=n+1;i<=m;i++)
                printf(" %d",map[i][0]);
            printf("\n");
        }
        else
        {
            printf("NO\n");
        }
    }
    return 0;
}