ACwing 851. spfa求最短路

给定一个 n 个点 m 条边的有向图，图中可能存在重边和自环， 边权可能为负数。

请你求出 1 号点到 n 号点的最短距离，如果无法从 1号点走到 n 号点，则输出 impossible。

数据保证不存在负权回路。

输入格式

第一行包含整数 n 和 m。

接下来 m 行每行包含三个整数 x,y,z，表示存在一条从点 x 到点 y 的有向边，边长为 z。

输出格式

输出一个整数，表示 1 号点到 n 号点的最短距离。

如果路径不存在，则输出 impossible。

数据范围

1≤n,m≤105,
图中涉及边长绝对值均不超过 10000。

输入样例：

3 3
1 2 5
2 3 -3
1 3 4

输出样例：

2

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;
const int maxn = 100100;
bool vis[maxn];
int dis[maxn],head[maxn];
struct node
{
    int v,next,w;
}e[maxn];
int cnt;
int step[maxn];
void add(int u,int v,int w)
{
    e[cnt].next=head[u];
    e[cnt].v=v;
    e[cnt].w=w;
    head[u]=cnt++;
}
void spfa(int n)
{
    memset(dis,inf,sizeof dis);
    dis[1]=0;
    queue<int>q;
    q.push(1);
    vis[1]=true;
    while(!q.empty())
    {
        int id = q.front();
        vis[id]=false;
        q.pop();
        for(int i = head[id];i!=-1;i=e[i].next)
        {
            int v = e[i].v;
            if(dis[v]>dis[id]+e[i].w)
            {
                dis[v]=dis[id]+e[i].w;
                step[v]=step[id]+1;
                if(step[v]>n-1)
                {
                    break;
                }
                if(!vis[v])
                {
                    q.push(v);
                    vis[v]=true;
                }
            }
        }
    }
    if(dis[n]>inf/2)cout<<"impossible"<<endl;
    else
    cout<<dis[n]<<endl;
}
int main()
{
    int n,m,u,v,w;
    memset(head,-1,sizeof head);
    cin>>n>>m;
    for(int i = 0;i<m;i++)
    {
        cin>>u>>v>>w;
        add(u,v,w);
    }
    spfa(n);
}