hdu 6386 Age of Moyu最短路

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6386

题解看了好久才看懂。。更新最短路径的方式是每个节点用一个set集合来维护。其中val代表到达此节点的当前最短路径。

pre为路径的上一个节点。相当于维护了不止一个信息，然后维护最短路径。如果更新了最短，就将set清空，否则加进去

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
#define mod 1000000007
#define For(i,m,n) for(int i=m;i<=n;i++)
#define Dor(i,m,n) for(int i=m;i>=n;i--)
#define LL long long
#define lan(a,b) memset(a,b,sizeof(a))
using namespace std;

set<int> sta[100010];

struct node
{
    int to,w,next;
}bian[100010*4];
int head[100010];

int tot=0;
void add(int p,int q,int w)
{
    bian[tot].to=q;
    bian[tot].w=w;
    bian[tot].next=head[p];
    head[p]=tot;
    tot++;
}
int dis[100010];


struct pou{
    int val,u;
    int pre,fa;
    bool operator <(const pou &p) const{return val>p.val;}
};
void pin(int s,int t)
{
    lan(dis,inf);
    dis[s]=0;
    priority_queue<pou> run;
    run.push((pou){dis[s],s,-1,-1});
    while(!run.empty())
    {
        pou x=run.top();
        run.pop();
        int pre=x.pre,u=x.u;
        if(x.val>dis[u])continue;
        else if(x.val==dis[u])
        {
            bool f=1;
            if(sta[u].find(pre)!=sta[u].end())
                continue;
            sta[u].insert(pre);
        }
        else
        {
            dis[u]=x.val;
            sta[u].clear();
            sta[u].insert(pre);
        }

        for(int i=head[u];~i;i=bian[i].next)
        {
            int to=bian[i].to,w=bian[i].w;
            if(to==x.fa)continue;
            if(dis[u]+(pre!=w)<=dis[to])
            {
                dis[to]=dis[u]+(pre!=w);
                if(to!=t)
                run.push((pou){dis[to],to,w,x.u});
            }
        }

    }

}


int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        tot=0;
        lan(head,-1);
        For(i,1,n)
        sta[i].clear();
        For(i,1,m)
        {
            int p,q,w;
            scanf("%d%d%d",&p,&q,&w);
            add(p,q,w);
            add(q,p,w);
        }
            pin(1,n);
            if(dis[n]==inf)
                dis[n]=-1;
            printf("%d\n",dis[n]);
    }
    return 0;
}