我居然看都沒看就做成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;
}