HK 算法 模板

View Code

int dist[N<<1],mx[N],my[N],m,n;
vector<int> map[N];
int que[N<<1],head,tail;
int bfs()
{
    int i;
    head=0;tail=-1;
    for(i=1;i<=n;i++)
        if(mx[i]==-1) que[++tail]=i;
        for(i=0;i<=m+n;i++) dist[i]=0;
        int flag=0;
        while(head<=tail)
        {
            int u=que[head++];
            for(i=0;i<map[u].size();i++)
            {
                int v=map[u][i];
                if(dist[n+v]==0)
                {
                    dist[n+v]=dist[u]+1;
                    if(my[v]!=-1)
                    {
                        dist[my[v]]=dist[n+v]+1;
                        que[++tail]=my[v];
                    }
                    else flag=1;
                }
            }
        }
        return flag;
}
int dfs(int u)
{
    for(int i=0;i<map[u].size();i++)
    {
        int v=map[u][i];
        if(dist[u]+1==dist[v+n])
        {
            int t=my[v];
            dist[v+n]=0;
            if(t==-1||dfs(t))
            {
                my[v]=u;
                mx[u]=v;
                return 1;
            }
        }
    }
    return 0;
}
int H_K()
{
    int i;
    for(i=0;i<=n;i++) mx[i]=-1;
    for(i=0;i<=m;i++) my[i]=-1;
    int ans=0;
    while(bfs())
    {
        for(i=1;i<=n;i++)
            if(mx[i]==-1&&dfs(i)) ans++;
    }
    return ans;
}