图结构练习——最短路径（Dijkstra优化模板）

图结构练习——最短路径
Description
给定一个带权无向图，求节点1到节点n的最短路径。

Input
输入包含多组数据，格式如下。
第一行包括两个整数n m，代表节点个数和边的个数。(n<=100)
剩下m行每行3个正整数a b c，代表节点a和节点b之间有一条边，权值为c。

Output
每组输出占一行，仅输出从1到n的最短路径权值。（保证最短路径存在）

Sample
Input
3 2
1 2 1
1 3 1
1 0
Output
1
0

#include <bits/stdc++.h>

using namespace std;
const int N = 1e5 + 10;
const int M = 1e9 + 7;
int h[505], cnt;
struct node {
    int w, v, next;
} a[N];
void add(int u, int v, int w) {
    a[++cnt].w = w;
    a[cnt].v = v;
    a[cnt].next = h[u];
    h[u] = cnt;
}
priority_queue<pair<int, int>> q;
int n, m, u, v, w;
int dis[N], vis[N];
void dijkstra() {
    memset(dis, 0x3f, sizeof(dis));
    memset(vis, 0, sizeof(vis));
    dis[1] = 0;
    q.push(make_pair(0, 1));
    while (!q.empty()) {
        int u = q.top().second;
        q.pop();
        if (vis[u]) continue;
        vis[u] = 1;
        for (int i = h[u]; i; i = a[i].next) {
            int v = a[i].v, w = a[i].w;
            if (dis[v] > dis[u] + w) {
                dis[v] = dis[u] + w;
                q.push(make_pair(-dis[v], v));
            }
        }
    }
}
int main() {
    ios::sync_with_stdio(false);
    while (cin>>n>>m) {
        cnt = 0;
        memset(h, 0, sizeof(h));
        for (int i = 0; i < m; i++) {
            cin >> u >> v >> w;
            add(u, v, w);
            add(v, u, w);
        }
        dijkstra();
        cout << dis[n] << '\n';
    }
    return 0;
}