题意:给你V个点,E条边N个队,已知N个队的位置,求最少K个队在不一样的位置时候的最短时间。
题解:spfa求出有队伍所在的地点对于其他地点的最短路,二分时间,用二分图匹配judge一下当前时间能否满足要求。
AC代码:
#include<stdio.h>
#include<queue>
#include<vector>
#include<string.h>
#include<algorithm>
#define inf 700000000
#define N 205
#define M 20005
using namespace std;
int pos[N],mark[605];
int dist[605][605];
int v,e,n,k;
struct edge
{
int to,next,w;
edge(){}
edge(int to,int next,int w)
{
this->to=to;
this->next=next;
this->w=w;
}
}ed[M*2];
int head[605],lnum;
vector<int>vt[605];
void addline(int a,int b,int w)
{
ed[lnum]=edge(b,head[a],w);
head[a]=lnum++;
}
void spfa(int u)
{
queue<int>que;
int mark[605];
memset(mark,0,sizeof(mark));
dist[u][u]=0;
mark[u]=1;
que.push(u);
while(!que.empty())
{
int k=que.front();
que.pop();
mark[k]=0;
for(int i=head[k];~i;i=ed[i].next)
{
int to=ed[i].to;
if(dist[u][to]>dist[u][k]+ed[i].w)
{
dist[u][to]=dist[u][k]+ed[i].w;
if(mark[to]==0)
{
mark[to]=1;
que.push(to);
}
}
}
}
}
bool g[205][605]; //邻接矩阵,true代表有边相连
bool visit[605]; //记录V2中的某个点是否被搜索过
int match[605]; //记录与V2中的点匹配的点的编号
// 匈牙利算法
bool dfs(int u)
{
for(int i=1;i<=v;++i)
{
if(g[u][i]&&!visit[i]) //如果节点i与u相邻并且未被查找过
{
visit[i]=true; //标记i为已查找过
if(match[i]==-1||dfs(match[i])) //如果i未在前一个匹配M中,或者i在匹配M中,但是从与i相邻的节点出发可以有增广路径
{
match[i]=u; //记录查找成功记录,更新匹配M(即“取反”)
return true; //返回查找成功
}
}
}
return false;
}
int hungary()
{
int ans=0;
memset(match,-1,sizeof(match));
for(int i=1;i<=n;i++)
{
memset(visit,false,sizeof(visit)); //清空上次搜索时的标记
if(dfs(i)) //从节点i尝试扩展
ans++;
}
return ans;
}
bool judge(int t)
{
memset(g,false,sizeof(g));
for(int i=1;i<=n;i++)
for(int j=1;j<=v;j++)
if(dist[pos[i]][j]<=t)
g[i][j]=true;
int sum=hungary();
return sum>=k;
}
int main()
{
memset(head,-1,sizeof(head));
lnum=0;
for(int i=0;i<605;i++)
for(int j=0;j<605;j++)
dist[i][j]=inf;
scanf("%d%d%d%d",&v,&e,&n,&k);
for(int i=1;i<=n;i++)
scanf("%d",&pos[i]);
for(int i=0;i<e;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
addline(u,v,w);
addline(v,u,w);
}
for(int i=1;i<=n;i++)
if(mark[pos[i]]==0)
{
mark[pos[i]]=1;
spfa(pos[i]);
}
int l=0,r=1731311;
while(l<=r)
{
int mid=l+r>>1;
if(judge(mid))r=mid-1;
else l=mid+1;
}
if(l>1731311)printf("-1\n");
else printf("%d\n",l);
}