洛谷 P2683 小岛

P2683 小岛

spfa，

#include<bits/stdc++.h>
using namespace std;
#define maxn 1000000
#define inf 1<<30
#define LL long long
int n,m,s,num,t,head[maxn],dis[maxn],if_;
struct Edge{
    int v,d,next;
}edge[maxn];

void ins(int u,int v,int d)
{
    edge[++num].v=v;
    edge[num].d=d;
    edge[num].next=head[u];
    head[u]=num;
}

void spfa(int s)
{
    queue<int>que;
    bool vis[maxn];
    memset(vis,false,sizeof(vis));
    for(int i=1;i<=n;i++) dis[i]=inf;
    dis[s]=0; vis[s]=1;
    que.push(s);
    while(!que.empty())
    {
        int cur=que.front();
        que.pop();
        for(int i=head[cur];i;i=edge[i].next)
        {
            if(dis[cur]+edge[i].d<dis[edge[i].v])
            {
                dis[edge[i].v]=dis[cur]+edge[i].d;
                if(!vis[edge[i].v])
                {
                    vis[edge[i].v]=1;
                    que.push(edge[i].v);
                }
            }
        }
        vis[cur]=false;
     } 
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
    {
        int u,v,d;
        scanf("%d",&if_);
        if(if_==1)
        {
            scanf("%d%d%d",&u,&v,&d);
            ins(u,v,d); ins(v,u,d);
        }
        else{
            scanf("%d%d",&s,&t);
            spfa(s);
            printf("%d\n",dis[t]==inf?-1:dis[t]);
        }
    }
    return 0;
}