3408: [Usaco2009 Oct]Heat Wave 热浪

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define INF 1000000000
int m,n,x,y,z,s,e;
int w[2505][2505];
int d[2505],vis[2505];
void dijkstra()
{
for(int i=1;i<=n;i++) d[i]=INF;
d[s]=0;
for(int i=1;i<=n;i++)
{
int x,m=INF;
for(int j=1;j<=n;j++)
{
if(m>d[j]&&!vis[j]){
m=d[j];
x=j;
}
}
vis[x]=1;
for(int j=1;j<=n;j++)
d[j]=min(d[j],d[x]+w[x][j]);
}
}
int main()
{
scanf("%d%d%d%d",&n,&m,&s,&e);
memset(w,63,sizeof(w));
memset(vis,0,sizeof(vis));
for(int i=1;i<=m;i++){
scanf("%d%d%d",&x,&y,&z);
if(z<w[x][y])
w[x][y]=w[y][x]=z;
}
dijkstra();
cout<<d[e]<<endl;
return 0;
}