很裸的二分匹配。。。
每个人可以到达第几把伞,然后建立二分图。。
不过,匈牙利算法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;
}