hdu 2389 Rain on your Parade

很裸的二分匹配。。。

每个人可以到达第几把伞，然后建立二分图。。

不过，匈牙利算法TLE，可以用H_K(Hopcroft_Karp)算法

 

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<queue>
using namespace std;
#define INF 10000000

struct Guest
{
    int x;
    int y;
    int sp;
} gst[3010];

queue<int> q;
int g[3010][3010];
int vis[3010];
int linkx[3010];
int linky[3010];
int distx[3010];
int disty[3010];
int m,n,t;
int dis;

bool CanReach(int k,int x,int y)
{
    int s= (gst[k].x-x)*(gst[k].x-x) + (gst[k].y-y)*(gst[k].y-y) ;
    if( s <= t*t*gst[k].sp*gst[k].sp )
        return 1;
    return 0;
}

bool findSearch()
{
    memset(distx,-1,sizeof(distx));
    memset(disty,-1,sizeof(disty));
    while(!q.empty()) q.pop();
    for(int i=0;i<m;i++)
        if(linkx[i]==-1)
        {
            q.push(i);
            distx[i]=0;
        }
    dis=INF;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        if(distx[u]>dis) break;
        for(int i=0;i<n;i++)
            if( disty[i]==-1 && g[u][i] )
            {
                disty[i]=distx[u]+1;
                if( linky[i]==-1 ) dis=disty[i];
                else
                {
                    distx[linky[i]]=disty[i]+1;
                    q.push(linky[i]);
                }
            }
    }
    return dis<INF;
}

bool findPath(int u)
{
    for(int i=0;i<n;i++)
        if( !vis[i] && g[u][i] && disty[i]==distx[u]+1 )
        {
            vis[i]=1;
            
            if( disty[i]==dis && linky[i]!=-1) 
                continue;
            if( linky[i]==-1 || findPath(linky[i]) )
            {
                linky[i]=u;
                linkx[u]=i;
                return 1;
            }
        }
    return 0;
}

int Hopcroft_Karp()
{
    int res=0;
    memset(linkx,-1,sizeof(linkx));
    memset(linky,-1,sizeof(linky));
    while(findSearch())
    {
        memset(vis,0,sizeof(vis));
        for(int i=0;i<m;i++)
            if(linkx[i]==-1)
                if( findPath(i) )
                    res++;
    }
    return res;
}

int main()
{
//    freopen("in.txt","r",stdin);
    int c;
    scanf("%d",&c);
    int ccc=1;
    while(c--)
    {
        scanf("%d %d",&t,&m);
        for(int i=0;i<m;i++)
        {
            int a,b,c;
            scanf("%d %d %d",&gst[i].x,&gst[i].y,&gst[i].sp);
        }
        scanf("%d",&n);
        memset(g,0,sizeof(g));
        for(int i=0;i<n;i++)
        {
            int x,y;
            scanf("%d %d",&x,&y);
            for(int j=0;j<m;j++)
                if(CanReach(j,x,y))
                    g[j][i]=1;
        }
        int ans=Hopcroft_Karp();
        printf("Scenario #%d:\n",ccc++);
        printf("%d\n\n",ans);
    }
    return 0;
}