HDU2389-Rain on your Parade（二分图匹配bfs模板）

每位顾客找伞，属于标准的二分图匹配问题，但是用深搜肯定妥妥的超时

存图也不用邻接矩阵了，用邻接表

#include<bits/stdc++.h>
using namespace std;

const int MAXN = 3010;//左边节点数量、右边节点数量
const int MAXM = 3010*3010;//边的数量
const int INF = 0x7FFFFFFF;

struct Edge
{
    int v;
    int next;
} edge[MAXM];

int nx, ny;
int cnt;
int dis;

int first[MAXN];
int xlink[MAXN], ylink[MAXN];//作用相当于match，xlink是人，ylink是伞
//xlink[i]表示左集合顶点所匹配的右集合顶点序号，ylink[i]表示右集合i顶点匹配到的左集合顶点序号。
int dx[MAXN], dy[MAXN];
//dx[i]表示左集合i顶点的距离编号，dy[i]表示右集合i顶点的距离编号
int vis[MAXN]; //寻找增广路的标记数组

struct point
{
    int x,y,v;
} a[MAXN],b[MAXN];


void init()
{
    cnt = 0;
    memset(first, -1, sizeof(first));
    memset(xlink, -1, sizeof(xlink));
    memset(ylink, -1, sizeof(ylink));
}

void read_graph(int u, int v)  //邻接表存图
{
    edge[cnt].v = v;
    edge[cnt].next = first[u], first[u] = cnt++;
}

int bfs()
{
    queue<int> q;
    dis = INF;//表示人与伞之间的距离
    memset(dx, -1, sizeof(dx));
    memset(dy, -1, sizeof(dy));
    for(int i = 0; i < nx; i++)
    {
        if(xlink[i] == -1)//如果人没有拿到伞
        {
            q.push(i);//那么把这个人放进栈里
            dx[i] = 0;
        }
    }
    while(!q.empty())
    {
        int u = q.front();
        q.pop();
        if(dx[u] > dis) break;
        for(int e = first[u]; e != -1; e = edge[e].next)
        {
            int v = edge[e].v;
            if(dy[v] == -1)
            {
                dy[v] = dx[u] + 1;
                if(ylink[v] == -1) //如果伞没有被取走
                    dis = dy[v];
                else
                {
                    dx[ylink[v]] = dy[v]+1;
                    q.push(ylink[v]);
                }
            }
        }
    }
    return dis != INF;
}

int find(int u)//u代表人
{
    for(int e = first[u]; e != -1; e = edge[e].next)//对于每个可匹配的伞
    {
        int v = edge[e].v;
        if(!vis[v] && dy[v] == dx[u]+1)
        {
            vis[v] = 1;
            if(ylink[v] != -1 && dy[v] == dis)
                continue;
            if(ylink[v] == -1 || find(ylink[v]))
            {
                xlink[u] = v, ylink[v] = u;
                return 1;
            }
        }
    }
    return 0;
}

int MaxMatch()
{
    int ans = 0;
    while(bfs())
    {
        memset(vis, 0, sizeof(vis));
        for(int i = 0; i < nx; i++)
            if(xlink[i] == -1)
            {
                ans += find(i);
            }
    }
    return ans;
}

int main()
{
    int T,t;
    int q=1;
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%d",&t);
        scanf("%d",&nx);
        for(int i=0; i<nx; i++)
        {
            scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].v);
        }
        scanf("%d",&ny);
        for(int i=0; i<ny; i++)
        {
            scanf("%d%d",&b[i].x,&b[i].y);
        }
        for(int i=0; i<nx; i++)
            for(int j=0; j<ny; j++)
            {
                if((a[i].x-b[j].x)*(a[i].x-b[j].x)+(a[i].y-b[j].y)*(a[i].y-b[j].y)<=t*a[i].v*t*a[i].v)
                    read_graph(i,j);
            }

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

    return 0;
}