初学算法之uva1599 IdealPath

 1 #include<cstdio>
 2 #include<vector>
 3 #include<queue>
 4 #include<cstring> 
 5 using namespace std; //min()函数 
 6 #define max 100000
 7 #define inf 10000000+10
 8 bool vis[max],in_q[max];
 9 int m,n,a,b,c;//边数、结点数、结点、下一个结点、颜色
10 int d[max];//记录各节点到终点的路径长度
11 int res[max];//记录最短路的权值
12 
13 typedef struct edge {
14     int pot, color;
15     edge(int v = 0, int c = 0) :pot(v), color(c){}
16 }Edge;//还有这种操作，我也是第一次用，赋值很方便
17 
18 vector<Edge>map[max];//记录邻接表
19 void re_bfs() {
20     int u, v;
21     memset(vis,0,sizeof(vis));
22     memset(in_q, 0, sizeof(in_q));
23     queue<int>q;
24     q.push(n - 1);
25     vis[n - 1] = true;
26     d[n - 1] = 0;
27 //在map里面存在着所有的节点以及他们之间的连接关系。map【u】.size代表着节点u相连节点的个数；
28     while (!q.empty()) {
29         u = q.front(); q.pop(); vis[u] = true;
30         for (int i = 0; i < map[u].size(); i++) {
31             v = map[u][i].pot;
32             if (!vis[v]&&!in_q[v]) {
33                 vis[v] = true;
34                 d[v] = d[u] + 1;
35                 q.push(v);
36                 in_q[v] = true;
37             }
38         }
39     }
40 }
41 
42 void bfs() {
43     memset(in_q, 0, n);
44     memset(vis, 0, sizeof(vis));
45     int u, v;
46     queue<int>q;
47     q.push(0);
48     memset(res, 0, n*sizeof(int));//res这个数组，存储了每个最短路径里面每条边的权值
49     while (!q.empty()) {
50         u = q.front(); q.pop(); vis[u] = true;
51         int minc = inf, num = map[u].size();
52         for (int i = 0; i < num; i++) {//这个循环是找出与节点u相连的边的最小权值
53             v = map[u][i].pot;
54             if (!vis[v] && d[u] - 1 == d[v])
55                 minc = min(map[u][i].color, minc);
56         }
57         for (int i = 0; i < num; i++) {//最小权值的边不一定是一个，所以把所对应的另一个结点压入队列
58             v = map[u][i].pot;
59             if (!vis[v] && d[u] - 1 == d[v] && map[u][i].color == minc && !in_q[v]) 
60                 q.push(v); in_q[v] = true;
61         }
62         int index = d[0] - d[u];//这里就有学问了，因为就不仅要求得最短的路径，还要满足这些边是相连的
63         if (res[index] == 0)res[index] = minc;
64         else {
65             if (minc > res[index]) {//这里把前一段权值相同的，而压入队列的几个结点pop出去较大的，留下最下的，如果实在不懂，就把（121）（131）（243）（352）（464）（565）画一个图就明白了。
66                 int temp_minc = q.front(); q.pop(); q.pop(); q.push(temp_minc);
67             }
68             else if (minc < res[index])q.pop();
69         }
70     }
71 }
72 
73 int main() {
74     while (scanf_s("%d%d", &n, &m) == 2) {
75         for (int j = 0; j < n; j++)map[j].clear();
76         memset(d, -1, sizeof(int)*n); 
77         while (m--) {
78             scanf_s("%d%d%d", &a, &b, &c);
79             if (a != b) {//我觉得in_q的判断已经去除了自环和重边的问题
80                 map[a - 1].push_back(edge(b - 1, c));
81                 map[b - 1].push_back(edge(a - 1, c));
82             }
83         }
84         re_bfs();
85         bfs();
86         printf("%d\n%d", d[0], res[0]);
87         for (int i = 1; i < d[0]; i++)printf("%3d", res[i]);
88         printf("\n");  
89     }
90 }