P3385 【模板】负环
P3385 【模板】负环 - 洛谷
题目描述
给定一个 n 个点的有向图，请求出图中是否存在从顶点 1 出发能到达的负环。
负环的定义是：一条边权之和为负数的回路。
输入格式
本题单测试点有多组测试数据。
输入的第一行是一个整数 T，表示测试数据的组数。对于每组数据的格式如下：
第一行有两个整数，分别表示图的点数 n 和接下来给出边信息的条数 m。
接下来 m 行，每行三个整数 u,v,w。
- 若 w>0，则表示存在一条从 u 至 v 边权为 w 的边，还存在一条从 v 至 u 边权为 w 的边。
- 若 w<0，则只表示存在一条从 u 至 v 边权为 w 的边。
输出格式
对于每组数据，输出一行一个字符串，若所求负环存在，则输出 YES，否则输出 NO。
输入输出样例
输入 #1
2
3 4
1 2 2
1 3 4
2 3 1
3 1 -3
2 3
1 2 3
2 3 4
3 1 -8输出 #1
NO
YES说明/提示
数据规模与约定
对于全部的测试点，保证：
- 1≤n≤2×103
- 1≤m≤3×103
- 1≤u,v≤n
- −104≤w≤104
- 1≤T≤10
提示
请注意，m 不是图的边数。
思路：
代码如下：
#include <iostream>
#include <algorithm>
#include <cmath>
#include <queue>
#include <cstring>
using namespace std;
typedef long long ll;
const ll N = 2e5 * 5 + 5;
ll tot = 0;
ll n, m, s;
struct Edge{
ll next,to,w;
}e[N];
ll head[N],dis[N];
void add(ll u,ll v,ll w)
{
tot++;
e[tot].next = head[u];
e[tot].to = v;
e[tot].w = w;
head[u] = tot;
}
void spfa()
{
queue<ll> q;
q.push(s);
memset(dis,0x3f,sizeof dis);
dis[s]=0;
while(!q.empty())
{
ll pos = q.front();
q.pop();
ll u = head[u];
while(u != -1)
{
ll to = e[u].to;
ll w = e[u].w;
if(dis[to] > w + dis[pos])
{
dis[to] = w + dis[pos];
q.push(to);
}
}
}
}
int main()
{
cin >> n >> m >> s;
for(ll i = 1 ; i <= m ; i++)
{
ll u,v,w;
cin >> u >> v >> w;
add(u,v,w);
}
spfa();
return 0;
}