SPFA算法

1.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<cstring>
#include<iostream>
#include<algorithm>
#include<queue>

using namespace std;

typedef pair<int,int> pii;

const int N = 1.5e5+5;
int h[N], w[N],idx, e[N], ne[N];
int n,m;
int dist[N];
bool st[N];
bool flag;

void add(int a, int b, int c){
    e[idx] = b,w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}

int spfa(){
    memset(dist,0x3f,sizeof dist);
    dist[1] = 0;
    queue<int> q;
    q.push(1);
    st[1] = true;
    while(!q.empty()){
        int t = q.front();
        q.pop();
        st[t] = false;
        for(int i=h[t];i!=-1;i=ne[i]){
            int j=e[i];
            if(dist[j]>dist[t]+w[i]){
                dist[j] = dist[t]+w[i];
                if(!st[j]){
                    q.push(j);
                    st[j] = true;
                }
            }
        }
    }
    if(dist[n]==0x3f3f3f3f) {
        flag = true;
        return -1;
    }
    return dist[n];
}
int main(){
    scanf("%d%d",&n,&m);
    memset(h,-1,sizeof h);
    while(m--){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    int t = spfa();
    if(t==-1 && flag) puts("impossible");
    else printf("%d\n",t);
    return 0;
}

2.spfa判断负环

给定一个 n个点 m条边的有向图，图中可能存在重边和自环， 边权可能为负数。
请你判断图中是否存在负权回路。
输入格式
第一行包含整数 n和 m。
接下来 m行每行包含三个整数 x,y,z，表示存在一条从点 x到点 y的有向边，边长为 z。
输出格式
如果图中存在负权回路，则输出 Yes，否则输出 No。
数据范围
1≤n≤2000,
1≤m≤10000,
图中涉及边长绝对值均不超过 10000。
输入样例：
3 3
1 2 -1
2 3 4
3 1 -4
输出样例：
Yes

#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>

using namespace std;

typedef pair<int,int> pii;

const int N = 1.5e5+5;
int h[N], w[N],idx, e[N], ne[N];
int n,m; 
int dist[N],cnt[N]; 
bool st[N]; 
bool flag; 

void add(int a, int b, int c){ 
    e[idx] = b,w[idx] = c, ne[idx] = h[a], h[a] = idx++; 
}

int spfa(){ 
    queue<int> q; 
    for(int i=1;i<=n;i++){// 由于判断负环不一定是从1号点开始的，所以从1号点开始的所有点进入队列中
        st[i] = true; 
        q.push(i); 
    }
    while(!q.empty()){ 
        int t = q.front(); 
        q.pop(); 
        st[t] = false; 
        for(int i=h[t];i!=-1;i=ne[i]){ 
            int j=e[i]; 
            if(dist[j]>dist[t]+w[i]){ 
                dist[j] = dist[t]+w[i]; 
                cnt[j] = cnt[t] + 1; 
                if(cnt[j]>=n) return true; 
                if(!st[j]){ 
                    q.push(j); 
                    st[j] = true; 
                }
            }
        }
    }
    return false;
}
int main(){
    scanf("%d%d",&n,&m);
    memset(h,-1,sizeof h);
    while(m--){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
    }
    bool t = spfa();
    if(t) puts("Yes");
    else puts("No");
    return 0;
}