求助
查看原帖
求助
541069
SuperCowHorse楼主2022/6/12 09:44

样例都过不了

#include<bits/stdc++.h>
using namespace std;
int t,n,m,u,v,w,cnt;
int head[6005],sum[2005],dis[2005];
bool vis[2005];
queue<int>q;
struct edge{
	int u,v,w,next;
}e[6005];
void add(int u,int v,int w)
{
	++cnt;
	e[cnt].u=u;
	e[cnt].v=v;
	e[cnt].w=w;
	e[cnt].next=head[u];
	head[u]=cnt;
}
bool spfa(int s)
{
	while(!q.empty()) q.pop();
	memset(dis,0x3f,sizeof(dis));
	memset(vis,0,sizeof(vis));
	memset(sum,0,sizeof(sum));
	dis[s]=0;vis[s]=1;++sum[s];
	q.push(s);
	while(!q.empty())
	{
		u=q.front();q.pop();
		vis[u]=0;
		for(int i=head[u];i;i=e[i].next)
		{
			v=e[i].v;
			if(dis[v]>dis[u]+e[i].w)
			{
				dis[v]=dis[u]+e[i].w;
				if(!vis[v])
				{
					vis[v]=1;
					if(++sum[v]>=n) return 1;
					q.push(v);
				}
			}
		}
	}
	return 0;
}
int main()
{
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d %d",&n,&m);
		for(int i=1;i<=n;++i)
		{
			scanf("%d %d %d",&u,&v,&w);
			if(w>=0)
				add(u,v,w),add(v,u,w);
			else
				add(u,v,w);
		}
		if(spfa(1)==1)
			printf("YES\n");
		else
			printf("NO\n");
	}
	return 0;
}
2022/6/12 09:44
加载中...