【PTA】7-10 Dijkstra算法（模板）（C++）

给一个n(1 ≤ n ≤ 2500) 个点 m(1 ≤ m ≤ 6200) 条边的无向图，求 s 到 t 的最短路。

输入格式:

第一行四个由空格隔开的整数 n、m、s、t。

之后的 m 行，每行三个正整数 si​、ti​、wi​(1≤wi​≤109)，表示一条从si​ 到 ti​ 长度为 wi​ 的边。

输出格式:

一个整数，表示从s 到t 的最短路径长度。数据保证至少存在一条道路。

输入样例:

7 11 5 4
2 4 2
1 4 3
7 2 2
3 4 3
5 7 5
7 3 3
6 1 1
6 3 4
2 4 3
5 6 3
7 2 1

输出样例:

7

注意:

两个顶点之间可能存在多条直接相连的道路。

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 2510;
int g[N][N];
int dist[N];
bool st[N];
int n,m;
int x,y;
void dijkstra()
{
    memset(dist , 0x3f ,sizeof dist);
    dist[x]=0;
    for(int i=0;i<n-1;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
            if(!st[j]&&(t==-1||dist[t]>dist[j]))
                t=j;
        for(int j=1;j<=n;j++)
            dist[j]=min(dist[j],g[t][j]+dist[t]);
        st[t]=true;
    }
}
int main()
{
    cin>>n>>m>>x>>y;
    memset(g , 0x3f , sizeof g);
    for(int i=0;i<m;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        g[a][b]=g[b][a]=min(g[a][b],c);
    }
    dijkstra();
    cout<<dist[y]<<endl;
    return 0;
}