2 1 1 2 3 3 3 1 2 5 2 3 5 3 1 2 0 0
3 2
第一次完成最短路径的题目,初探Dijkstra算法的精深。
思路就是通过Dijkstra算法找到1到其余各个路口的最短距离,自然就找到1到n的最短距离。输出即可。
原本我使用的不是正确的Dijkstra算法,但依然通过了很多组数据,后来才发现不对,从而对Dijkstra算法有了更深入的理解。
WA code:
#include <iostream> #include <cstdio> using namespace std; int main() { const int N=101,max=10000000; int n,m,i,j,cost[N],path[N][N],a,b,c;//cost记录各个路口到1的距离,path记录各个路口间的距离 while(scanf("%d%d",&n,&m) && n>=1) { for(i=1;i<=n;i++) { for(j=1;j<=i;j++) { path[i][j]=path[j][i]=max; } path[i][i]=0; } for(i=2;i<=n;i++) { cost[i]=max; } cost[1]=0; for(i=0;i<m;i++) { scanf("%d%d%d",&a,&b,&c); path[a][b]=path[b][a]=c; if(a==1) cost[b]=c; else if(b==1) cost[a]=c; } //错误之处: //================================================================== for(i=2;i<n;i++) { if(cost[i]<max) { for(j=2;j<=n;j++) { if(cost[j]>path[i][1]+path[i][j]) { path[j][1]=path[1][j]=cost[j]=path[i][1]+path[i][j]; } } } } //=================================================================== printf("%d\n",cost[n]); } return 0; }
5
1 2 4
2 4 1
1 3 1
3 2 2
1 4 5
AC code:
#include <iostream> #include <cstdio> using namespace std; int main() { const int N=101,max=0xfffff;//max不宜设得过大,避免后面的运算溢出 //cost记录各个路口到1的距离,path记录各个路口间的距离 int n,m,i,j,k,cost[N],path[N][N],a,b,c,min; bool S[N]; while(scanf("%d%d",&n,&m) && n>=1) { //初始化 for(i=1;i<=n;i++) { for(j=1;j<=i;j++) { path[i][j]=path[j][i]=max; } path[i][i]=0; } for(i=2;i<=n;i++) { S[i]=0; cost[i]=max; } S[1]=1; cost[1]=0; //输入 for(i=0;i<m;i++) { scanf("%d%d%d",&a,&b,&c); path[a][b]=path[b][a]=c; if(a==1) cost[b]=c; else if(b==1) cost[a]=c; } //从路口2开始到路口n-1即可 for(i=2;i<n;i++) { min=max; //寻找最短路径 for(j=2;j<=n;j++) { if(!S[j] && cost[j]<min) { min=cost[j]; k=j; } } S[k]=1;//把当前离路口1最近的路口纳入集合 //更新路径 for(j=2;j<=n;j++) { if(!S[j] && cost[j]>path[k][j]+min) { cost[j]=path[k][j]+min; } } } printf("%d\n",cost[n]); } return 0; }