题解：AT_past202109_h 最短経路

思路

这一眼就是最短路的题目啊。

为什么不用 Dijkstra 用 死了的 SPFA。因为好写 。

这一题的数据比较小。可以暴力枚举最短路的起点，跑 SPFA，找到符合的直接输出，结束程序。随机数据下 SPFA 平均的时间复杂度为 $O(kn)$。那么总的时间复杂度为 $O(k{n}^2)$，可以通过此题。需要注意的是，题目是无向边。

这随机的数据下，SPFA 是快一点的，毕竟是 $O(kn)$。

代码

#include<bits/stdc++.h>
#include<cstring>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#include<map>
#define ll long long
using namespace std;
const int N=5e5+10;
const int M=2147483647;
const int inf=0x3f3f3f3f;
int n,m,s,dis[N];
int x,y,z;
vector<int> a[N],b[N];
void bfs(int s)
{
	memset(dis,inf,sizeof dis);
	queue<int> q;
	q.push(s);
	dis[s]=0;
	while(q.size())
	{
		int now=q.front();
		q.pop();
		for(int i=0;i<a[now].size();i++)
		{
			int xx=a[now][i];
			if(dis[now]+b[now][i] <dis[xx])
			{
				q.push(xx);
				dis[xx]=dis[now]+b[now][i];
			}
		}
	}
} 
int main()
{
	scanf("%d%d",&n,&m); 
	for(int i=1;i<n;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		a[x].push_back(y);
		b[x].push_back(z);
		a[y].push_back(x);
		b[y].push_back(z);
	}
	for(int i=1;i<=n;i++)
	{
		bfs(i);
		for(int j=1;j<=n;j++)
		{
			if(dis[j]==m)
			{
				printf("Yes");
				return 0;
			}
		}
	}
	printf("No");
	return 0;
}