最简单的免费网站制作模板跨境电商营销推广
思路
这一眼就是最短路的题目啊。
为什么不用 Dijkstra
用 死了的 SPFA
。因为好写 。
这一题的数据比较小。可以暴力枚举最短路的起点,跑 SPFA,找到符合的直接输出,结束程序。随机数据下 SPFA 平均的时间复杂度为 O ( k n ) O(kn) O(kn)。那么总的时间复杂度为 O ( k n 2 ) O(kn^2) O(kn2),可以通过此题。需要注意的是,题目是无向边。
这随机的数据下,SPFA
是快一点的,毕竟是 O ( k n ) O(kn) 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;
}