HDU 2389 Rain on your Parade(HK算法)

#include<iostream>
#include<cstring>
#include<queue>
#include<cmath>
using namespace std;
const int maxn=3000+5;
struct guest{int x,y,s;}guests[maxn];
struct umbrella{int x,y;}umbrellas[maxn];
int t,m,n;
const int inf=(1<<30);
bool line[maxn][maxn];
int cx[maxn],cy[maxn];
int nx,ny;
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<=nx;i++){
if(cx[i]==-1) {q.push(i);dx[i]=0;}//对于未遍历的点入队
}
//准备分层
while(!q.empty())
{
int u=q.front();
q.pop();
if(dx[u]>dis) break;//如果目前的层次大于最小增广长度，那么退出
for(int j=1;j<=ny;j++)//对于一切可能的点遍历
{
if(line[u][j]==true&&dy[j]==-1){//只对未分层的点遍历
dy[j]=dx[u]+1;
if(cy[j]==-1) dis=dy[j];
else{
dx[cy[j]]=dy[j]+1;
q.push(cy[j]);
}
}
}
}
return dis!=inf;
}
//寻找路径配对
bool findpath(int x)
{
for(int j=1;j<=ny;j++)
{
if(!used[j]&&line[x][j]&&dy[j]==dx[x]+1)//符合继续搜索的条件有三个：未访问过，图上联通和层次符合
{
used[j]=1;
if(cy[j]!=-1&&dis==dy[j]) continue;//如果下一个点还是匹配点且目前已经到达增广最小层次，不需要扩展了
if(cy[j]==-1||findpath(cy[j]))
{
cy[j]=x;cx[x]=j;
return true;
}
}
}
return false;
}
int HK()
{
int ans=0;
memset(cx,-1,sizeof(cx));
memset(cy,-1,sizeof(cy));
while(searchP())
{
memset(used,0,sizeof(used));
for(int i=1;i<=nx;i++){
if(cx[i]==-1)
{
if(findpath(i)) ans++;
}
}
}
return ans;
}
double get_t(guest g,umbrella u)
{
int xx=g.x-u.x;
int yy=g.y-u.y;
double dis=sqrt(xx*xx+yy*yy);
return dis/g.s;
}
int main()
{
int T;
scanf("%d",&T);
for(int i=1;i<=T;i++)
{
memset(line,false,sizeof(line));
scanf("%d",&t);
scanf("%d",&m);
for(int i=1;i<=m;i++) {scanf("%d%d%d",&guests[i].x,&guests[i].y,&guests[i].s);}
scanf("%d",&n);
for(int i=1;i<=n;i++) {scanf("%d%d",&umbrellas[i].x,&umbrellas[i].y);}
for(int i=1;i<=m;i++)
{
for(int j=1;j<=n;j++)
{
if(get_t(guests[i],umbrellas[j])<=t) line[i][j]=true;
}
}
nx=m;ny=n;
printf("Scenario #%d:\n%d\n\n",i,HK());
}
return 0;
}