【JZOJ】【匈牙利算法】【二分】导弹

导弹

有一个节点为$t$的无向图，分为AB两个部分，A部分有$n$个节点，B部分有$m$个节点($n$>=$m$）（可能有一些节点不属于A部分，也不属于B部分），现在A部分的节点要到B部分的每一个节点，问，最少要多少时间？

样例输入

3
0 2 1
2 0 10
1 10 0
1
2
1
3

样例输出

3

前言

这一道题有关匈牙利算法，大家可以先了解一下匈牙利算法

思路

先求出一个点到另一个点的最短路，然后二分答案，每一次二分时跑一次匈牙利

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
struct whw
{
	int x,y,t;
}wh[2005];
int f[2005][2005],r[2005],c[2005],h[2005],hw[2005];
int n,m,tot,ans;
bool flog[2005];
void check(int x,int y)//连边
{wh[++tot]=(whw){x,y,h[x]};h[x]=tot;}
bool Hungarian(int k)//匈牙利算法
{
	int t=0;
	for(int i=h[k];i;i=wh[i].t)
	{
		if(!flog[wh[i].y])
		{
			t=hw[wh[i].y];
			hw[wh[i].y]=k;
			flog[wh[i].y]=1;
			if(!t || Hungarian(t))return 1;//如果可以连就退出
			hw[wh[i].y]=t;
		}
	}
	return 0;
}
int Bs(int k)
{
	memset(h,0,sizeof(h));
	memset(hw,0,sizeof(hw));
	ans=0,tot=0;
	for(int i=1;i<=n;++i)
		for(int j=1;j<=m;++j)
			if(f[r[i]][c[j]]<=k)check(i,j);//连边
	for(int i=1;i<=n;++i)
	{
		memset(flog,0,sizeof(flog));
		if(Hungarian(i))ans++;//A城的每一个城市都有发射导弹到达的B城城市
	}
	return ans==m;
}
int main()
{
	freopen("c.in","r",stdin);
	freopen("c.out","w",stdout);
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
			scanf("%d",&f[i][j]);
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
			for(int k=1;k<=n;++k)
				f[i][j]=min(f[i][j],f[i][k]+f[k][j]);//我这里用的是Floyed//跑一遍最短路
	scanf("%d",&n);
	for(int i=1;i<=n;++i)
		scanf("%d",&r[i]);
	scanf("%d",&m);
	for(int i=1;i<=m;++i)
		scanf("%d",&c[i]);
	int l=1,r=1e9;
	while(l<=r)//二分答案
	{
		int mid=(l+r)/2;
		if(Bs(mid))r=mid-1;
		else l=mid+1;
	}
	printf("%d",l);
	fclose(stdin);
	fclose(stdout);
	return 0;
}