【codevs 1557】热浪(Dijkstra)

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
int n,m,ts,te;
const int maxn=200005;
int first[maxn],next[maxn<<1],tot=0;
int dis[maxn];
bool vis[maxn];
struct edge
{
    int f,t,v;
}es[maxn];
struct zt
{
    int u,v;
    bool operator <(const zt &b)const 
    {
        return v>b.v;
    } 
};
priority_queue<zt> q;
void build(int f,int t, int v)
{
    es[++tot]=(edge){f,t,v};
    next[tot]=first[f];
    first[f]=tot;
    return;
}
void dij(int s,int t)
{
    memset(vis,0,sizeof(vis));
    memset(dis,0x3f,sizeof(dis));
    while(!q.empty())
        q.pop();
    dis[s]=0;
    q.push((zt){s,0});
    while(!q.empty())
    {
        zt x=q.top();
        q.pop();
        int u=x.u;
        if(u==t) return ;
        if(vis[u]) continue;
        vis[u]=1;
        for(int i=first[u];i!=-1;i=next[i])
        {
            int w=es[i].t;
            if(dis[w]>dis[u]+es[i].v)
            {
                dis[w]=dis[u]+es[i].v;
                q.push((zt){w,dis[w]});
            }
        }
    }
    return;
}
int main()
{
    memset(first,0xff,sizeof(first));
    scanf("%d%d%d%d",&n,&m,&ts,&te);
    int a,b,c;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        build(a,b,c);
        build(b,a,c);
    }
    dij(ts,te);
    printf("%d\n",dis[te]);
    return 0;
}