HDU2389 Rain on your Parade（HK模版）

题意：

在一个二维坐标系上有n个人和m把伞，每个人都有自己的移动速度，

问有多少人可以在s min内移动到不同的雨伞处（不允许两个人共用一把伞）。

思路：

由于nm太大（3000），匈牙利会超时，就用HK算法

整理了HK的模版

/* ***********************************************
//Author        :devil
//Created Time  :2016/5/10 23:13:51
//************************************************ */
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <stdlib.h>
using namespace std;
const int N=3010;
const int inf=0x3f3f3f3f;
vector<int>eg[N];
int ax[N],ay[N],as[N],bx[N],by[N],n,m;
int mx[N],my[N];
int dx[N],dy[N],dis;
bool vis[N];
bool bfs()
{
    queue<int>q;
    dis=inf;
    memset(dx,-1,sizeof(dx));
    memset(dy,-1,sizeof(dy));
    for(int i=0;i<n;i++)
        if(mx[i]==-1)
        {
            q.push(i);
            dx[i]=0;
        }
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        if(dx[u]>dis) break;
        for(int i=0;i<eg[u].size();i++)
        {
            int v=eg[u][i];
            if(dy[v]==-1)
            {
                dy[v]=dx[u]+1;
                if(my[v]==-1) dis=dy[v];
                else
                {
                    dx[my[v]]=dy[v]+1;
                    q.push(my[v]);
                }
            }
        }
    }
    return dis!=inf;
}
bool dfs(int u)
{
    for(int i=0;i<eg[u].size();i++)
    {
        int v=eg[u][i];
        if(!vis[v]&&dy[v]==dx[u]+1)
        {
            vis[v]=1;
            if(my[v]!=-1&&dy[v]==dis) continue;
            if(my[v]==-1||dfs(my[v]))
            {
                my[v]=u;
                mx[u]=v;
                return 1;
            }
        }
    }
    return 0;
}
int MaxMatch()
{
    int ans=0;
    memset(mx,-1,sizeof(mx));
    memset(my,-1,sizeof(my));
    while(bfs())
    {
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++)
            if(mx[i]==-1&&dfs(i))
                ans++;
    }
    return ans;
}
int main()
{
    //freopen("in.txt","r",stdin);
    int t,s,cas=0;
    scanf("%d",&t);
    while(t--)
    {
        for(int i=0;i<=3000;i++)
            eg[i].clear();
        scanf("%d%d",&s,&n);
        for(int i=0;i<n;i++)
            scanf("%d%d%d",&ax[i],&ay[i],&as[i]);
        scanf("%d",&m);
        for(int i=0;i<m;i++)
            scanf("%d%d",&bx[i],&by[i]);
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                if((ax[i]-bx[j])*(ax[i]-bx[j])+(ay[i]-by[j])*(ay[i]-by[j])<=as[i]*as[i]*s*s)
                    eg[i].push_back(j);
        printf("Scenario #%d:\n",++cas);
        printf("%d\n\n",MaxMatch());
    }
    return 0;
}