个人模板 最短路djkstra

在n个点m条边的图中求s点到e点的最短距离

#include<bits/stdc++.h>
#define inf 1e7
using namespace std;
int m,n,s,e;
int a[101][101];
int dis[101];
int vis[101];
void djkstra(int m,int k)
{
    for(int i=1; i<=n; i++)             ///初始化
    {
        for(int j=1; j<=n; j++)
        {
            if(i==j)
                a[i][j]=0;
            else
                a[i][j]=inf;
        }
    }
    int t1,t2,t3;
    for(int i=0; i<m; i++)              ///建立双向图
    {
        scanf("%d%d%d",&t1,&t2,&t3);
        a[t1][t2]=t3;
        a[t2][t1]=t3;
    }
    for(int i=1; i<=n; i++)             ///dis数组记录K点到各个点间距离
    {
        dis[i]=a[s][i];
        vis[i]=0;
    }
    vis[1]=1;
    for(int i=1,u; i<n; i++)            ///松弛操作
    {
        int mmin=inf;
        for(int j=1; j<=n; j++)
        {
            if(!vis[j]&&dis[j]<mmin)
            {
                mmin=dis[j];
                u=j;
            }
        }
        vis[u]=1;
        for(int k=1; k<=n; k++)
        {
            if(a[u][k]<inf)
            {
                if(dis[k]>a[u][k]+dis[u])
                    dis[k]=a[u][k]+dis[u];
            }
        }
    }
}
int main()
{
    while(scanf("%d%d%d%d",&n,&m,&s,&e)==4)
    {
        djkstra(m,s);
        printf("%d\n",dis[e]);
    }
}