ZOJ 2504 Help John! (dijkstra)

我感覺這個題的敘述有問題。

 

john他媽給他說了一條路線，但是他想走更短的。但是他走出去的第一條路必須是她媽說的，剩下的就隨便走了。

他如果找不到可以到達的路的話，就輸出N。

 

數據也很弱，我的程序漏洞百出也能過。

 

#include<stdio.h> #include<math.h> const int INF = 99999990; int t,n,r,w,m; int from,to,now; int mat[1001][1001]; int mo[30001]; int dist[1001],flag[1001]; int i,j,k,start = 1; int sumself,summo,min; int main(void) { scanf("%d",&t); for(k = 1; k <= t; k++ ) { scanf("%d%d",&n,&r); for(i = 1; i <= n; i++ ) //初始化 { dist[i] = INF; flag[i] = 0;//把0寫成1了，wa死了 for( j = 1; j <= n;j++ ) mat[i][j]=INF; } for(i = 1; i <= r; i++ ) //輸入矩陣 { scanf("%d%d%d",&from,&to,&w); mat[from][to] = mat[to][from] = w; } scanf("%d",&m); //他媽給的路線長度 for( i = 1; i <= m; i++ ) scanf("%d",&mo[i]); for( i = 1,summo = 0; i < m; i++ ) summo += mat[ mo[i] ][ mo[i+1] ]; dist[mo[2]] = mat[1][mo[2]]; //從第二個到n點的最短路徑 flag[now = mo[2]] = 1; flag[1] = 1; for ( i = 2; i < n; ++ i ) { for( j = 2; j <= n; ++ j ) if ( !flag[j] && dist[j] > dist[now] + mat[now][j] ) dist[j] = dist[now] + mat[now][j]; for( j = 2,min = INF; j <= n; j++ ) if ( !flag[ j ] && min > dist[j] ) min = dist[ now=j ]; flag[now] = 1; } sumself = dist[n]; printf("TEST %d ",start++); if( sumself == INF ) printf("N/n"); else printf("Y %d/n",summo-sumself); } return 0; }