【ACWing】852. spfa判断负环

题目地址:

https://www.acwing.com/problem/content/854/

给定一个 n n n个点 m m m条边的有向图,图中可能存在重边和自环, 边权可能为负数。请你判断图中是否存在负权回路。

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

输出格式:
如果图中存在负权回路,则输出“Yes”,否则输出“No”。

数据范围:
1 ≤ n ≤ 2000 1\le n\le 2000 1n2000
1 ≤ m ≤ 10000 1\le m\le 10000 1m10000
0 ≤ ∣ e ∣ ≤ 10000 0\le |e|\le 10000 0e10000
e e e是边长

SPFA算法可以找负环,基本思路是在更新某个点的最短路的时候(这里”某个点的最短路“并没有说源点是谁,其实我们考虑的是每个点都要作为源点去尝试一下。也可以考虑成有个假想的超级源点,这个源点与每个点的距离都是 0 0 0,那么第一次是将超级源点入队,第二次就会将所有顶点入队了),看看到达该点的最短路的边数是否多于 n n n。这里的距离数组dist是不用初始化的,一开始就是 0 0 0,表示自己到自己的最短路是 0 0 0(或者表示这些点到超级源点的距离是 0 0 0)。如果某个点的最短路径边数大于等于 n n n,代表存在一条有重复顶点的路径到达它比那个重复顶点不走,路程反而更短,这就说明存在负环。SPFA找负环一般要把队列换成栈,这样更快。代码如下:

#include <cstring>
#include <iostream>
#include <stack>
using namespace std;

const int N = 100010;
int n, m;
int h[N], w[N], e[N], ne[N], idx;
int dist[N], cnt[N];
bool st[N];

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

bool spfa() {
  stack<int> stk;
  for (int i = 1; i <= n; i++) {
    stk.push(i);
    st[i] = true;
  }

  while (stk.size()) {
    int t = stk.top();
    stk.pop();

    st[t] = false;
    for (int i = h[t]; ~i; 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]) {
          stk.push(j);
          st[j] = true;
        }
      }
    }
  }

  return false;
}

int main() {
  memset(h, -1, sizeof h);
  cin >> n >> m;
  while (m--) {
    int a, b, c;
    cin >> a >> b >> c;
    add(a, b, c);
  }

  spfa() ? puts("Yes") : puts("No");
}

时间复杂度 O ( m n ) O(mn) O(mn),空间 O ( n + m ) O(n+m) O(n+m)

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值