易错点:
- 由于是普通队列,不需要判断Node的大小关系,所以可以记录当前点是否在队列中(vis[]),如果在则不需要重复入队.
- 判断队列前端和后端哪个dis更小,让更小的在前面(swap(q.front(),q.back()))是一个有效剪枝.
- 数据清空时邻接表(head[])必须清空,但边只需要重置edgeCnt即可.
#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int MAXN=4e3,MAXM=6e3;
struct Edge{
int from,to,w,nxt;
}e[MAXM];
int head[MAXN],edgeCnt=0;
void addEdge(int u,int v,int w){
e[++edgeCnt].from=u;
e[edgeCnt].to=v;
e[edgeCnt].w=w;
e[edgeCnt].nxt=head[u];
head[u]=edgeCnt;
}
int n,dis[MAXN],cnt[MAXN];
bool vis[MAXN];
bool spfa(){
queue<int> q;
q.push(1);
dis[1]=0;
while(!q.empty()){
if(dis[q.front()]>dis[q.back()])swap(q.front(),q.back());
int u=q.front();q.pop();
vis[u]=0;
for(int i=head[u];i;i=e[i].nxt){
int nowV=e[i].to;
if(dis[nowV]>dis[u]+e[i].w){
dis[nowV]=dis[u]+e[i].w;
cnt[nowV]=cnt[u]+1;
if(cnt[nowV]>n)return 1;
if(!vis[nowV]){
vis[nowV]=1;
q.push(nowV);
}
}
}
}
return 0;
}
void init(){
memset(dis,0x3f,sizeof(dis));
memset(cnt,0,sizeof(cnt));
memset(vis,0,sizeof(vis));
memset(head,0,sizeof(head));//
edgeCnt=2;
}
int main(){
int t;
scanf("%d",&t);
while(t--){
init();
int m;
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++){
int a,b,w;
scanf("%d%d%d",&a,&b,&w);
addEdge(a,b,w);
if(w>=0)addEdge(b,a,w);
}
cout<<(spfa()?"YE5":"N0")<<endl;
}
return 0;
}