求助自认为没错但re+wa,顺便讨论一下是大于n还是大于等于n和初始点入队算不算
查看原帖
求助自认为没错但re+wa,顺便讨论一下是大于n还是大于等于n和初始点入队算不算
183990
隋乐珉楼主2020/12/1 22:52
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+500; 
int first[maxn],nxt[maxn],to[maxn],v[maxn];
int cnt;
int n,m;
int cntt[maxn];
int dis[maxn];
bool vis[maxn];
int q[maxn];
bool ok;
void add(int x,int y,int z){
	nxt[++cnt]=first[x];
	first[x]=cnt;
	to[cnt]=y;
	v[cnt]=z;
}
void init(){
	cnt=0;
	ok=0;
	memset(q,0,sizeof(q));
	memset(nxt,0,sizeof(nxt));
	memset(first,0,sizeof(first));
	memset(dis,0x3f,sizeof(dis));
	memset(vis,0,sizeof(vis));
	memset(cntt,0,sizeof(cntt));
}
void spfa(int s)
{
	dis[s]=0;
	cntt[s]=1;
	int l=1;
	int r=1;
	q[l]=s;
	vis[s]=1;
	while(l<=r)
	{
		int now=q[l++]; 
		for(int i=first[now];i;i=nxt[i])
		{
			int ver=to[i];
			if(dis[ver]>dis[now]+v[i])
			{
			//	cerr<<dis[ver]<<" "<<ver<<endl;
			//	cout<<dis[now]+v[i]<<endl;
				dis[ver]=dis[now]+v[i];
				if(!vis[ver]){
					q[++r]=ver;
					vis[ver]=1;
					cntt[ver]++;	
					//cerr<<ver<<"   "<<cntt[ver]<<endl; 
					if(cntt[ver]>n) 
		{
		ok=1;
		return;
		}
					}
			}
		}
	
		vis[now]=0;
	}
	
}
int main(){
	int T;
	cin>>T;
	while(T--)
	{
		init();
		scanf("%d%d",&n,&m);
		for(int i=1;i<=m;i++)
		{
			int x,y,vs;
			scanf("%d%d%d",&x,&y,&vs);
			if(vs<0)add(x,y,vs);
			else
				add(x,y,vs),
				add(y,x,vs); 
		}
		spfa(1);
		if(ok==1)
		printf("YES\n");	
		else 
		printf("NO\n");
	}
	return 0;
}
2020/12/1 22:52
加载中...