Snuke's Subway Trip

http://icpc.upc.edu.cn/problem.php?cid=1700&pid=7

#include<cstdio>
#include<set>
#include<queue>
#include<iostream>
#include<cstring>
using namespace std;
struct str{
    int t,n,w;
}e[1000000];
int t,i,h[10000000],u,vv,w,v[10000000],d[10000000],n,m;
set<int>c[1000000];
void add(int u,int v,int w)
{
    t++;
    e[t].t=v;
    e[t].w=w;
    e[t].n=h[u];
    h[u]=t;
}
priority_queue<pair<int,int> >q;
void dij()
{
    memset(d,0x3f,sizeof(d));
    d[1]=0;
    q.push(make_pair(d[1],1));
    while (q.size())
    {
        int u=q.top().second;
        q.pop();
        if (v[u])
            continue;
        v[u]=1;
        for (int i=h[u]; i; i=e[i].n)
        {
            int k=e[i].t,w=c[u].count(e[i].w)?0:1;
            if (d[k]>d[u]+w)
            {
                d[k]=d[u]+w;
                q.push(make_pair(-d[k],k));
                c[k].clear();
                c[k].insert(e[i].w);
            }
            else if (d[k]==d[u]+w)
                c[k].insert(e[i].w);
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for (i=1; i<=m; i++)
    {
        scanf("%d%d%d",&u,&vv,&w);
        add(u,vv,w);
        add(vv,u,w);
    }
    dij();
    if (d[n]==0x3f3f3f3f)
        printf("-1\n");
    else
        printf("%d\n",d[n]);
}

猜你喜欢

转载自blog.csdn.net/weixin_43601103/article/details/88376271