6 2 3 1 3 5 1 4 7 2 8 12 3 8 4 4 9 12 9 10 2 1 2 8 9 10
9
第一道floyd算法写的最短路,看到同学用SPFA写的0ms代码,终于深刻体会到floyd的时间复杂度
需要继续努力学习更多算法
#include <cstdio> #include <algorithm> #include <cstring> using namespace std; const int INF=0x3f3f3f3f; int w[1005][1005],ans,index[1005],num,sta[1005],des[1005],stas,dess; int i,j,k,t,c,d,s,e,si,ei; int main() { while(scanf("%d%d%d",&t,&c,&d)==3) { memset(index,0,sizeof(index)); memset(w,0x3f,sizeof(w)); ans=INF,num=stas=dess=0; for(i=0;i<t;++i) { w[i][i]=0; scanf("%d%d%d",&s,&e,&k); if(!index[s]) index[s]=++num; if(!index[e]) index[e]=++num; if(w[si=index[s]][ei=index[e]]>k) w[si][ei]=w[ei][si]=k; } while(c--) { scanf("%d",&s); if(index[s]) sta[stas++]=index[s]; } while(d--) { scanf("%d",&e); if(index[e]) des[dess++]=index[e]; } for(k=0;k<num;++k) for(i=0;i<num;++i) if(w[i][k]!=INF)//floyd算法耗时太大,不加就TLE,加上后变为249ms for(j=0;j<i;++j)//由于是无向图,可以只算左斜下的,加上后变为93ms if((s=w[i][k]+w[k][j])<w[i][j]) w[i][j]=w[j][i]=s; for(i=0;i<stas;++i) for(j=0;j<dess;++j) ans=min(ans,w[sta[i]][des[j]]); printf("%d\n",ans); } return 0; }