POJ 1719 Shooting Contest

/* 给定一个 r*c 的格子矩阵，有C发子弹，每列格子有2个白色的方块，在1，2，3，...c 列上开枪打白色的格子，打到一个，那么这个白色方块所在的行就算打到了，求怎样打，才能把所有的行全部打到，如果全部打到而且有多余的列，那么这列随便打一个方块就可以了，按列的1,2,3,4......C的顺序，输出要打的行号。很明显的一个二分匹配，普通的二分匹配算法就行了。*/

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <memory.h>
using namespace std;

#define INF 0x7ffffff
#define MAXN 4000
#define MAXE 20000

struct Edge{
    int u,v;
    int next;
}edge[MAXE];
int net[MAXN],visit[MAXN],pre[MAXN];
int nv,ne,index,r,c;
void add_edge(const int& u, const int& v)
{
    edge[index].next = net[u];
    net[u] = index;
    edge[index].v = v;
    edge[index].u = u;
    ++index;
}
int dfs(const int& u)
{
    int i,v;
    for(i = net[u]; i != -1; i = edge[i].next)
    {
        v = edge[i].v;
        if(!visit[v])
        {
            visit[v] = 1;
            if(-1 == pre[v] || dfs(pre[v]))
            {
                pre[v] = u;
                return 1;
            }
        }
    }
    return 0;
}
int main()
{
    int i,j,k,h,tmp,sum,time,tmp1,tmp2,ans;
    scanf("%d",&time);
    while(time--)
    {
        scanf("%d%d",&r,&c);
        index = ans = 0;
        memset(net,-1,sizeof(net));
        memset(pre,-1,sizeof(pre));
        for(i = 1;i <= c; ++i)
        {
            scanf("%d%d",&tmp1,&tmp2);
            add_edge(tmp1,i);
            add_edge(tmp2,i);
            add_edge(i+r,tmp1);
        }
        if(c<r)
            ans = -1;
        else
            for(i = 1;i <= r; ++i)
            {
                memset(visit,0,sizeof(visit));
                if(dfs(i))
                    ++ans;
            }
        if(ans < r)
            printf("NO\n");
        else
        {
            for(i = 1;i <= c; ++i)
            {
                if(i != 1) printf(" ");
                if(pre[i] != -1)
                    printf("%d",pre[i]);
                else
                    printf("%d",edge[net[i+r]].v);
            }
            printf("\n");
        }
    }
    return 0;
}