dijkstra最短路入门模板题
poj 2387Til the Cows Come Home
题目翻译:
兔子很热爱学习,为了多学习她希望可以找最快的路线去上学。她的城市里有N个(2 <= N <= 1000)个地铁站,编号分别为1…N。兔子家在1号地铁站旁边,学校是N号地铁站。地铁站之间共有M (1 <= M <= 2000)条双向路径。兔子现在在1号地铁站,她希望知道到学校最短需要多少时间。可以保证兔子能到达学校。忽略换乘地铁的等待时间
Input
第一行:m和n
接下来的m行:每一行有x,y,z三个整数。代表x站到y站或者y站到x站需要z分钟(1<=z<=100)
Output
兔子到学校最少需要多少时间。
Sample Input
5 5
1 2 20
2 3 30
3 4 20
4 5 20
1 5 100
Sample Output
90
题目大意:
给n个点m条边(注意:对于一对点可能给多种weight的边,取最小)兔子从1出发走到n,求单源点最短路径的长度。
定义变量:
maps[][]存图,初始值0x3f3f3f3f
dist[j] 表示到j点的最短路径长,初始值0x3f3f3f3f
vis[]访问过的点标记,再遇到时跳过
主要过程:
dijkstra算法有点类似于prim算法
先定一个源点,本题是1,dist[1] = 0即可从1出发
从1向外“松弛”更新所有的点
代码是:(t此时等于1)
for(int j=1;j<=n;j++){
dist[j] = min(dist[j],dist[t]+maps[t][j]);
}
膨胀过,就把vis[1]=true,防止再从1出发
之后找到下一个距离最近的点t,再进行上述过程。
核心思想还是贪心,每次找距离最近的点松弛,松弛完再找最近的点再松弛,只需循环n-1次就能更新所有的点。
#include<iostream>
#include<cmath>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#include<cstring>
//#include<bits/stdc++.h>
using namespace std;
const int nmax=2000,mmax=2010;
int n,m;
int maps[nmax][nmax];
int dist[nmax];
bool vis[nmax];
void dijkstra(){
dist[1] = 0;
for(int i=0;i<n-1;i++){
int t = -1;
for(int j = 1;j<=n;j++){
if(!vis[j]&&(t==-1||dist[j]<dist[t]))
t = j;
}
vis[t] = true;
for(int j=1;j<=n;j++){
dist[j] = min(dist[j],dist[t]+maps[t][j]);
}
}
}
int main(){
scanf("%d%d",&m,&n);
memset(maps,0x3f,sizeof(maps));
memset(dist,0x3f,sizeof(dist));
memset(vis,false,sizeof(vis));
for(int i=0;i<m;i++){
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
if(maps[u][v]>w){
maps[u][v]=w;
maps[v][u]=w;
}
}
dijkstra();
cout<<dist[n];
}