poj 1719 Shooting Contest (匈牙利算法)

Posted by 111qqz on Monday, May 30, 2016

TOC

poj1719题目链接

题意:射箭比赛,靶子是一个n*m的网格。网格的特点是没列只有两个白色,剩下的全是黑色。一共射m次,每列射一次,要求每行都射到至少一次才算合法,问是否有合法射法,如果有输出一组解。

思路:由于列数可能比行数多,那么我们把行编号作为左集合,列编号作为右集合,做一次匈牙利,对于每一行都找到它匹配的列,如果匹配数等于n,说明有解。对于输出一组街的问题,m列中已经匹配了n列,输出他们对应的link即可。对于剩下的m-n列,可以任意输出。由于之前不知道会剩下哪n-m列,所以在读入的时候有必要保存每一列的white格子的位置信息。

1A,有点爽。

/* ***********************************************
Author :111qqz
Created Time :2016年05月30日 星期一 20时44分59秒
File Name :code/poj/1719.cpp
************************************************ */

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <cstdlib>
#include <ctime>
#define fst first
#define sec second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ms(a,x) memset(a,x,sizeof(a))
typedef long long LL;
#define pi pair < int ,int >
#define MP make_pair

using namespace std;
const double eps = 1E-8;
const int dx4[4]={1,0,0,-1};
const int dy4[4]={0,-1,1,0};
const int inf = 0x3f3f3f3f;
const int N=1E3+7;
int n,m;
char maze[N][N];
int head[2*N];
int cnt;
int link[2*N];
int vis[2*N];
pi white[N];
struct Edge
{
    int v;
    int nxt;
}edge[N*2];

void addedge( int u,int v)
{
    edge[cnt].v = v;
    edge[cnt].nxt = head[u];
    head[u] = cnt;
    cnt++;
}
void init()
{
    cnt =  0;
    ms(head,-1);

}


int find( int u)
{
    for ( int i = head[u] ; i!=-1 ; i = edge[i].nxt)
    {
    int v = edge[i].v;
    if (vis[v]) continue;
    vis[v] = true;
    if (link[v]==-1||find(link[v]))
    {
        link[v] = u ;
        return true;
    }
    }
    return false;
}
int hungary()
{
    int res = 0 ;
    ms(link,-1);

    for ( int i = 1 ; i <= n ; i++)
    {
    ms(vis,false);
    if (find(i)) res++;
    }
  //  cout<<"res:"<<res<<endl;
    return res;
}
int main()
{
    #ifndef  ONLINE_JUDGE 
    freopen("code/in.txt","r",stdin);
  #endif
    int T;
    cin>>T;
    while (T--)
    {
        init();
        scanf("%d%d",&n,&m);
        int tot = 0;
        for ( int i = 1 ; i <= m ; i++)
        {
        int x1,x2;
        scanf("%d %d",&x1,&x2);
        white[++tot]=make_pair(x1,x2);

        addedge(x1,i+n);
        addedge(x2,i+n);
        }

        int ans = hungary();
        if (ans<n)
        {
        puts("NO");
        }
        else
        {
        //for ( int i = n+1 ; i <= n+m ; i++) cout<<"link[i]:"<<link[i]<<endl;
        for ( int i = n+1 ; i <= n+m ; i++)
        {
            if (link[i]==-1)
            {
            printf("%d ",white[i-n].fst);
            }
            else
            {
            printf("%d ",link[i]);
            }
        }
        printf("\n");
        }

    }

  #ifndef ONLINE_JUDGE  
  fclose(stdin);
  #endif
    return 0;
}

「真诚赞赏,手留余香」

111qqz的小窝

真诚赞赏,手留余香

使用微信扫描二维码完成支付