请编写程序求给定正权有向图的单源最短路径长度。图中包含n个顶点,编号为0至n-1,以顶点0作为源点。
输入格式:
输入第一行为两个正整数n和e,分别表示图的顶点数和边数,其中n不超过20000,e不超过1000。接下来e行表示每条边的信息,每行为3个非负整数a、b、c,其中a和b表示该边的端点编号,c表示权值。各边并非按端点编号顺序排列。
输出格式:
输出为一行整数,为按顶点编号顺序排列的源点0到各顶点的最短路径长度(不含源点到源点),每个整数后一个空格。如源点到某顶点无最短路径,则不输出该条路径长度。
输入样例:
1 2 3 4 5
| 4 4 0 1 1 0 3 1 1 3 1 2 0 1
|
输出样例:
思路:
- 这里采取的是邻接表存储信息的操作,因为n值较大,设置dis数组存储源点到各个点的最短距离,注意初始化
- 采用了最普通的Dijkstra算法,每次寻找最小权值,并判断是否可以进行松弛。
- 最后输出时注意,无最短路径时(即两点间距离为无穷时),不输出对应的dis值
代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
| #include <iostream> #include <vector> using namespace std;
typedef struct { int v, w; } Node;
const int inf = 65535; int n, e; int dis[20001]; vector<Node> v[20001]; bool vis[20001] = {false};
void dijkstra(int s) { fill(dis, dis + 20001, inf); dis[0] = 0; for (int i = 0; i < n; i++) { int u = -1, minx = inf; for (int j = 0; j < n; j++) { if (!vis[j] && dis[j] < minx) { u = j; minx = dis[j]; } } if (u == -1) break; vis[u] = true; for (int j = 0; j < v[u].size(); j++) { int x = v[u][j].v; if (!vis[x] && dis[u] + v[u][j].w < dis[x]) { dis[x] = dis[u] + v[u][j].w; } } } }
int main() { scanf("%d %d", &n, &e); for (int i = 0; i < e; i++) { int a, b, c; scanf("%d %d %d", &a, &b, &c); v[a].push_back(Node{b, c}); } dijkstra(0); for (int i = 1; i < n; i++) { if (dis[i] != inf) printf("%d ", dis[i]); } return 0; }
|