HDU2389 Rain on your Parade —— 二分图最大匹配 HK算法

题目链接：https://vjudge.net/problem/HDU-2389

 

Rain on your Parade

Time Limit: 6000/3000 MS (Java/Others)    Memory Limit: 655350/165535 K (Java/Others)
Total Submission(s): 4889    Accepted Submission(s): 1612

Problem Description

You’re giving a party in the garden of your villa by the sea. The party is a huge success, and everyone is here. It’s a warm, sunny evening, and a soothing wind sends fresh, salty air from the sea. The evening is progressing just as you had imagined. It could be the perfect end of a beautiful day.
But nothing ever is perfect. One of your guests works in weather forecasting. He suddenly yells, “I know that breeze! It means its going to rain heavily in just a few minutes!” Your guests all wear their best dresses and really would not like to get wet, hence they stand terrified when hearing the bad news.
You have prepared a few umbrellas which can protect a few of your guests. The umbrellas are small, and since your guests are all slightly snobbish, no guest will share an umbrella with other guests. The umbrellas are spread across your (gigantic) garden, just like your guests. To complicate matters even more, some of your guests can’t run as fast as the others.
Can you help your guests so that as many as possible find an umbrella before it starts to pour? 

Given the positions and speeds of all your guests, the positions of the umbrellas, and the time until it starts to rain, find out how many of your guests can at most reach an umbrella. Two guests do not want to share an umbrella, however. 

 

 

Input

The input starts with a line containing a single integer, the number of test cases.
Each test case starts with a line containing the time t in minutes until it will start to rain (1 <=t <= 5). The next line contains the number of guests m (1 <= m <= 3000), followed by m lines containing x- and y-coordinates as well as the speed si in units per minute (1 <= s_i <= 3000) of the guest as integers, separated by spaces. After the guests, a single line contains n (1 <= n <= 3000), the number of umbrellas, followed by n lines containing the integer coordinates of each umbrella, separated by a space.
The absolute value of all coordinates is less than 10000.

 

 

Output

For each test case, write a line containing “Scenario #i:”, where i is the number of the test case starting at 1. Then, write a single line that contains the number of guests that can at most reach an umbrella before it starts to rain. Terminate every test case with a blank line.

 

 

Sample Input

2 1 2 1 0 3 3 0 3 2 4 0 6 0 1 2 1 1 2 3 3 2 2 2 2 4 4

 

 

Sample Output

Scenario #1: 2 Scenario #2: 2

 

 

Source

HDU 2008-10 Public Contest

 

 

Recommend

lcy

 

 

题解：

就直接求二分图最大匹配，不过由于数据较大，匈牙利算法超时，所以需要用HK算法。

 

 

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <sstream>
#include <algorithm>
using namespace std;
const int INF = 2e9;
const int MOD = 1e9+7;
const int MAXN = 3000+10;

struct Node
{
    int x, y, speed;
}gue[MAXN], umb[MAXN];

int uN, vN, t;
vector<int>g[MAXN];

int Mx[MAXN], My[MAXN];
int dx[MAXN], dy[MAXN];
int dis;
bool used[MAXN];

bool SearchP()
{
    queue<int>Q;
    dis = INF;
    memset(dx, -1, sizeof(dx));
    memset(dy, -1, sizeof(dy));
    for(int i = 1; i<=uN; 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;
        int sz = g[u].size();
        for(int i = 0; i<sz; i++)
        {
            int v = g[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)
{
    int sz = g[u].size();
    for(int i = 0; i<sz; i++)
    {
        int v = g[u][i];
        if(!used[v] && dy[v]==dx[u]+1)
        {
            used[v] = true;
            if(My[v]!=-1 && dy[v]==dis) continue;
            if(My[v]==-1 || DFS(My[v]))
            {
                My[v] = u;
                Mx[u] = v;
                return true;
            }
        }
    }
    return false;
}

int MaxMatch()
{
    int res = 0;
    memset(Mx, -1, sizeof(Mx));
    memset(My, -1, sizeof(My));
    while(SearchP())
    {
        memset(used, false, sizeof(used));
        for(int i = 1; i<=uN; i++)
            if(Mx[i]==-1 && DFS(i))
                res++;
    }
    return res;
}

int main()
{
    int T, kase = 0;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d%d", &t, &uN);
        for(int i = 1; i<=uN; i++)
        {
            scanf("%d%d%d", &gue[i].x, &gue[i].y, &gue[i].speed);
            g[i].clear();
        }

        scanf("%d", &vN);
        for(int i = 1; i<=vN; i++)
            scanf("%d%d", &umb[i].x, &umb[i].y);

        for(int i = 1; i<=uN; i++)
        for(int j = 1; j<=vN; j++)
        {
            int dis = (gue[i].x-umb[j].x)*(gue[i].x-umb[j].x)
                      +(gue[i].y-umb[j].y)*(gue[i].y-umb[j].y);
            int s = gue[i].speed*gue[i].speed*t*t;
            if(s>=dis) g[i].push_back(j);
        }

        int ans = MaxMatch();
        printf("Scenario #%d:\n%d\n\n", ++kase, ans);

    }
}