我居然看都没看就做成Floyd 求多源最短路了,但很容易超时,要注意一下细节,毕竟o(n^3)
做完看了巨巨是我题解,把草儿家看成0,草儿到邻近的城市花费的时间看成0,这样就变成了从0到各个点的单源最短路问题,这样o(n^2)就可以过了
要学习大佬们的思维方式,嗯!
Floyd:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int T,S,D,ii;
const int N=1005;
const int inf=0x3f3f3f3f;
int mp[N][N],maxn,minn;
bool src[N],ed[N];
void init()
{
int a,b,t;
memset(mp,inf,sizeof(mp));
memset(src,false,sizeof(src));
memset(ed,false,sizeof(ed));
minn=inf;
for(int i=1; i<=T; i++)
{
cin>>a>>b>>t;
maxn=maxn<a?a:maxn;
maxn=maxn<b?b:maxn;
mp[a][b]=mp[b][a]=mp[a][b]>t?t:mp[a][b];
}
for(int i=1; i<=S; i++)
{
cin>>ii;
src[ii]=true;
}
for(int i=1; i<=D; i++)
{
cin>>ii;
ed[ii]=true;
}
}
void Floyd()
{
for(int k=1; k<=maxn; k++)
for(int i=1; i<=maxn; i++)
if(mp[i][k]!=inf)//减少
for(int j=1; j<=maxn; j++)
{
mp[i][j]=min(mp[i][j],mp[i][k]+mp[k][j]);
if(src[i]&&ed[j])
minn=min(minn,mp[i][j]);
}
cout<<minn<<endl;
}
int main()
{
while(scanf("%d%d%d",&T,&S,&D)!=EOF)
{
init();
Floyd();
}
return 0;
}