poj 3255

次短路练习

spfa+A*

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#define MAX 1e9
#define MAXE 200100
using namespace std;
struct edge
{
    int v,w,n;
}e[MAXE];

int n,m;
int head[10000],num,dis[10000],in[10000],rdis[10000];
void add(int a,int b,int c)
{
    e[num].v=b;
    e[num].w=c;
    e[num].n=head[a];
    head[a]=num;
    num++;
}

void SPFA(int k,int d[])
{
    for(int i=1;i<=n;i++)
        d[i]=MAX;
    d[k]=0;
    memset(in,0,sizeof in);
    queue <int> q;
    q.push(k);
    in[k]=1;
    while(!q.empty())
    {
        int t;
        t=q.front();
        q.pop();
        in[t]=0;
        for(int i=head[t];i!=-1;i=e[i].n)
        {
            int v=e[i].v;
            int w=e[i].w;
            if(d[v]>d[t]+w)
            {
                d[v]=d[t]+w;
                if(in[v]==0)
                {
                    in[v]=1;
                    q.push(v);
                }
            }
        }
    }
}

int main()
{
//    freopen("in","r",stdin);
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(head,-1,sizeof head);
        num=0;
        for(int i=0;i<m;i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
            add(b,a,c);
        }
        SPFA(1,dis);
        SPFA(n,rdis);
        int ans=MAX;
        for(int i=1;i<=n;i++)
            for(int j=head[i];j!=-1;j=e[j].n)
            {
                int v=e[j].v;
                int w=e[j].w;
                int temp=dis[i]+rdis[v]+w;
                if(temp>dis[n]&&ans>temp)
                    ans=temp;
            }
        printf("%d\n",ans);
    }
    return 0;
}