HDU2389 Rain on your Parade 二分图匹配Hopcroft-Karp算法

这道题很明显的二分图匹配,但是用匈牙利算法果断TLE。。。

网上一搜才发现还有个叫Hopcroft-Karp算法的东西emmmm...

附上模板博客:http://www.cnblogs.com/penseur/archive/2013/06/16/3138981.html

附上AC代码:

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int INF=0x3f3f3f3f;
typedef pair<double,double>pp;

const int MAXN=3005;
int g[MAXN][MAXN];

int mx[MAXN];//mx[i]表示左集合i顶点所匹配的右集合的顶点序号
int my[MAXN]; //my[i]表示右集合i顶点所匹配的左集合的顶点序号

int nx,ny;
int dx[MAXN];
int dy[MAXN];
int dis;
bool vis[MAXN];
bool searchpath()
{
    queue<int>Q;
    dis=INF;
    memset(dx,-1,sizeof(dx));
    memset(dy,-1,sizeof(dy));
    for(int i=0;i<nx;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 v=0;v<ny;v++)
       {
          if(g[u][v]&&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;
}

int findpath(int u)
{
    for(int v=0;v<ny;v++)
    {
       if(!vis[v]&&g[u][v]&&dy[v]==dx[u]+1)
       {
          vis[v]=1;
          if(my[v]!=-1&&dy[v]==dis)
          {
             continue;
          }
          if(my[v]==-1||findpath(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(searchpath())
    {
       memset(vis,0,sizeof(vis));
       for(int i=0;i<nx;i++)
       {
          if(mx[i]==-1)
          {
             ans+=findpath(i);
          }
       }
    }
    return ans;
}


int n,m;
double sp[MAXN];
pp gu[MAXN],um[MAXN];
double time;
int main()
{
    int t;
    scanf("%d",&t);
    for(int tt=1;tt<=t;tt++)
    {
        scanf("%lf",&time);
        scanf("%d",&m);
        for(int i=0;i<m;i++)
        {
            scanf("%lf%lf%lf",&gu[i].first,&gu[i].second,&sp[i]);
        }
        scanf("%d",&n);
        for(int i=0;i<n;i++)
            scanf("%lf%lf",&um[i].first,&um[i].second);
        memset(g,0,sizeof(g));
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
            {
                double d=sqrt((gu[i].first-um[j].first)*(gu[i].first-um[j].first)+(gu[i].second-um[j].second)*(gu[i].second-um[j].second));
                if((d/sp[i])<=time)
                    g[i][j]=1;
            }
        nx=m,ny=n;
        int ans=MaxMatch();
        printf("Scenario #%d:\n",tt);
        printf("%d\n\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Cc_Sonia/article/details/81974561