个人模板 最短路djkstra

#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]);
}
}