蓝桥杯 道路和航路（SPFA算法求最短路径）

//蓝桥杯 道路和航路  （SPFA算法求最短路径）//超时的两组数据#include<iostream>#include<cstdio>#include<string.h>#include<queue>using namespace std;const int N=30000;const int Inf=100000000;struct node{    int to;    int value;    int next;};int head[N];int vis[N];//int dis[N]={0};//存取距离int t,r,P,s;node p[N*6];void init(){    memset(head,-1,sizeof(head[0])*(t+2));    memset(vis,0,sizeof(vis[0])*(t+2));    int k=0;    int a,b,c;    for(int i=0;i<r;i++)    {        scanf("%d%d%d",&a,&b,&c);        p[k].to=b;        p[k].value=c;        p[k].next=head[a];        head[a]=k++;        p[k].to=a;        p[k].value=c;        p[k].next=head[b];        head[b]=k++;    }    for(int i=0;i<P;i++)    {        scanf("%d%d%d",&a,&b,&c);        p[k].to=b;        p[k].value=c;        p[k].next=head[a];        head[a]=k++;    }}//============关键算法===================bool relax(int pr,int to,int value){    if(dis[to]>dis[pr]+value)    {        dis[to]=dis[pr]+value;        return true;    }    return false;}int SPAF(int d){    for(int i=0;i<=t;i++)    {        dis[i]=Inf;    }    vis[d]=1;    dis[d]=0;    queue<int> que;    que.push(d);    while(!que.empty())    {        int pre=que.front();        que.pop();        vis[pre]=0;        for(int i=head[pre];i+1;i=p[i].next)        {            if(relax(pre,p[i].to,p[i].value)&&!vis[p[i].to])            {                que.push(p[i].to);                vis[p[i].to]=1;            }        }    }}//==========================================int main(){    scanf("%d%d%d%d",&t,&r,&P,&s);    init();    SPAF(s);    for(int i=1;i<=t;i++)    {        if(dis[i]==Inf)        {            printf("NO PATH\n");        }        else        {            printf("%d\n",dis[i]);        }    }    return 0;}