牛客网 2018年全国多校算法寒假训练营练习比赛(第四场) F.Call to your teacher-最短路(Dijkstra)
F.Call to your teacher
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld
题目描述
从实验室出来后,你忽然发现你居然把自己的电脑落在了实验室里,但是实验室的老师已经把大门锁上了。更糟的是,你没有那个老师的电话号码。你开始给你知道的所有人打电话,询问他们有没有老师的电话,如果没有,他们也会问自己的同学来询问电话号码。那么,你能联系到老师并且拿到电脑吗。
输入描述:
存在多组测试样例
每组样例的第一行分别是两个整数n(1<n<=50),m(1<m<=2000),n是在题目当中出现的人数,其中你的序号是1号,实验室老师的序号是n。
接下来的m行,每行有两个整数x(1<=x<=n),y(1<=y<=n),代表x有y的电话号码。
输出描述:
对于每组测试样例,如果你最终能联系到老师,输出“Yes”,否则输出“No”。
示例1
输入
5 5 1 3 2 3 3 4 2 4 4 5
输出
Yes
示例2
输入
4 3 1 2 2 3 4 1
输出
No
这个题迪杰斯特拉乱跑一下就可以。
代码:
1 #include<iostream>
2 #include<cstdio>
3 #include<cstring>
4 #include<cmath>
5 #include<queue>
6 #include<set>
7 #include<algorithm>
8 #include<map>
9 using namespace std;
10 #define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
11 const int maxn=2000+10;
12 const int inf=100000;
13 int n,bian;
14 int mp[maxn][maxn];
15 int u,v,cost;
16 int dis[maxn];
17 int pre[maxn];
18 void init(){
19 ios;
20 cin>>n>>bian;
21 for(int i=1;i<=n;i++)
22 for(int j=1;j<=n;j++)
23 mp[i][j]=inf;
24 for(int i=0;i<bian;i++){
25 cin>>u>>v;
26 cost=1;
27 mp[u][v]=mp[v][u]=cost;
28 }
29 for(int i=1;i<=n;i++)
30 dis[i]=inf;
31 }
32 void dijkstra(int n,int v){
33 bool vis[maxn];
34 for(int i=1;i<=n;i++){
35 vis[i]=false;
36 dis[i]=mp[v][i];
37 if(dis[i]==inf)
38 pre[i]=0;
39 else pre[i]=v;
40 }
41 dis[v]=0;
42 vis[v]=true;
43 for(int i=2;i<=n;i++){
44 int u=v;
45 int mazz=inf;
46 for(int j=1;j<=n;j++){
47 if(!vis[j]&&dis[j]<mazz){
48 u=j;
49 mazz=dis[j];
50 }
51 }
52 vis[u]=true;
53 for(int k=1;k<=n;k++){
54 if(!vis[k]&&mp[u][k]<inf){
55 if(dis[k]>mp[u][k]+dis[u]){
56 dis[k]=mp[u][k]+dis[u];
57 pre[k]=u;
58 }
59 }
60 }
61 }
62 }
63 int main(){
64 ios;
65 while(cin>>n>>bian){
66 for(int i=1;i<=n;i++)
67 for(int j=1;j<=n;j++)
68 mp[i][j]=inf;
69 for(int i=0;i<bian;i++){
70 cin>>u>>v;
71 cost=1;
72 mp[u][v]=cost;
73 }
74 for(int i=1;i<=n;i++)
75 dis[i]=inf;
76 dijkstra(n,1);
77 if(dis[n]!=inf)
78 cout<<"Yes"<<endl;
79 else
80 cout<<"No"<<endl;
81 }
82 return 0;
83 }
84