话说这题本来书上说用floyd就能写出来了,但是我写的只能在POJ上过,ZOJ过不了,实在想不出别的方法,最后问了瓜神的思路
先跑一遍floyd,求得各个交叉点到已有的消防站的最小距离的最大值;再对每个非消防站的路口进行一次spfa,求出假设此交叉点已经为消防站时各点到消防站
距离的最小距离的最大值,和原来的最大值比较,如果比原来的最大值小,更新最大值,并记录这个交叉点;
PS:0x7f真不是个好东西,就是这个害我wa了近2个小时,一直调不出来,最后改成0xfffffff就能AC了
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <climits>
#include <string>
#include <iostream>
#include <map>
#include <cstdlib>
#include <list>
#include <set>
#include <queue>
#include <stack>
using namespace std;
#define INF 0xfffffff
int dis[555],dist[5555];
int edge[555][555];
int m;
void floyd()
{
int i,j,k;
for(k=1;k<=m;k++)
{
for(i=1;i<=m;i++)
{
for(j=1;j<=m;j++)
{
edge[i][j]=min(edge[i][j],edge[i][k]+edge[k][j]);
}
}
}
}
void spfa(int s)
{
int vis[555];
memset(vis,0,sizeof(vis));
for(int i=1;i<=m;i++)
dist[i]=dis[i];
dist[s]=0;
queue<int>q;
q.push(s);
vis[s]=1;
while(!q.empty())
{
int u=q.front();
q.pop();
vis[u]=0;
for(int i=1;i<=m;i++)
{
if(dist[i]>dist[u]+edge[u][i])
{
dist[i]=dist[u]+edge[u][i];
if(!vis[i])
{
vis[i]=1;
q.push(i);
}
}
}
}
}
int main()
{
int n,a,b,c,i,j;
int vis[555],val[555];
char str[1000];
while(cin>>n>>m)
{
memset(vis,0,sizeof(vis));
memset(val,0,sizeof(val));
for(i=1;i<=m;i++)
dis[i]=INF;
for(i=1;i<=n;i++)
{
cin>>val[i];
vis[val[i]]=1;
dis[val[i]]=0;
}
for(i=1;i<=m;i++)
{
for(j=1;j<=m;j++)
{
if(i==j) edge[i][j]=0;
else edge[i][j]=INF;
}
}
getchar();
while(gets(str)&&strlen(str))
{
sscanf(str,"%d%d%d",&a,&b,&c);
edge[a][b]=edge[b][a]=c;
}
floyd();
for(i=1;i<=n;i++)
{
for(j=1;j<=m;j++)
{
if(vis[j]) continue;
dis[j]=min(dis[j],edge[val[i]][j]);
}
}
for(i=1;i<=n;i++)
{
dis[val[i]]=0;
}
int MAX=0;
for(i=1;i<=m;i++)
MAX=max(MAX,dis[i]);
int ans=1;
for(i=1;i<=m;i++)
{
if(vis[i]) continue;
spfa(i);
int MAX1=0;
for(j=1;j<=m;j++)
{
if(vis[j]) continue;
MAX1=max(MAX1,dist[j]);
}
if(MAX1<MAX)
{
MAX=MAX1;
ans=i;
}
}
cout<<ans<<endl;
}
return 0;
}