求助,spfa代码66分
查看原帖
求助,spfa代码66分
844651
hzy20100509楼主2023/7/20 15:13
#include<bits/stdc++.h>
using namespace std;
const int maxn=3001;
int n,m,idx,to[maxn],head[maxn],ne[maxn],w[maxn],dis[maxn],vis[maxn];
bool v[maxn];
queue<int> q;
void add(int x,int y,int z){
    to[++idx]=y;
    w[idx]=z;
    ne[idx]=head[x];
    head[x]=idx;
} 
bool spfa(int x){
	dis[x]=0;
	q.push(x);
	v[x]=1;
	while(!q.empty()){
		int now=q.front();
		q.pop();
		v[now]=false;
	
		for(int i=head[now];i;i=ne[i]){
			int val=to[i];
			if(dis[val]>dis[now]+w[i]){
				dis[val]=dis[now]+w[i];
				vis[val]++;
				if(vis[val]>=n) return 1;
				if(!v[val]){
					q.push(val);
					v[val]=1;
				}
			}
		}
	}
	return 0;
}
int main(){
	ios::sync_with_stdio(false);
	int t;cin>>t;
	for(int i=1;i<=t;i++){
		idx=0;
		while(!q.empty()) q.pop();
		memset(to,0,sizeof(to));
		memset(head,0,sizeof(head));
		memset(ne,0,sizeof(ne));
		memset(w,0,sizeof(w));
		memset(dis,0x3f,sizeof(dis));
		memset(v,0,sizeof(v));
		memset(vis,0,sizeof(vis));
		cin>>n>>m;
		for(int i=1;i<=m;i++){
			int a,b,c;
			cin>>a>>b>>c;
			if(c>=0) add(b,a,c);
			add(a,b,c);
		}
		if(spfa(1)) cout<<"YES"<<endl;
		else cout<<"NO"<<endl;
	}
	return 0;
}

66分,Tle了3个,Wa一个。 望大佬帮助。

2023/7/20 15:13
加载中...