廣度優先搜索練習之神奇的電梯
發表時間:2020-10-19
發布人:葵宇科技
浏覽次數:27
Description
有一座已知層數為n的高樓,這座高樓的特殊之處在于隻能靠電梯去上下(xià)樓,所以要去到某一層要非常耽誤時間,然而更悲哀的是,這座高樓的電梯是限号的,小鑫最開始的時候在1層,他想去第x層,問(wèn)題是他最起碼要經過多少(shǎo)層(包含第x層)才能到達第x層。
Input
多組輸入。
第一行是三個(gè)正整數n,m,q。分别代表樓的總層數,給定的m條信息和(hé)q次查詢。
接下(xià)來的m行,每行的第一個(gè)整數pos代表這是第pos層的電梯,第二個(gè)數代表從這一層可(kě)以去的樓層總共有num個(gè),之後的num個(gè)數字代表從第pos層代表可(kě)以去的樓層。
最後的q行,每行一個(gè)整數代表小鑫想去的樓層号碼。
1<=m,pos,num<=n<=200
1<=q<=20
Output
對于每次詢問(wèn)輸出一個(gè)整數,占一行。代表如(rú)果要去某個(gè)樓層最少(shǎo)要經過多少(shǎo)層,如(rú)果到不了的話就輸出-1。
Sample
Input
10 4 3
1 2 6 7
3 4 4 6 8 10
5 2 2 3
7 3 10 5 6
4
5
9
Output
5
3
-1
這是AC代碼,後面有一個(gè)wrong answer代碼,求大佬指點
#include<bits/stdc++.h>
using namespace std;
int a[300][300];
int visit[300];
int dis[300];
int n,m,k;
int t;
void bfs(int x)
{
visit[x]=1;
dis[x]=0;
queue<int>q;
q.push(x);
while(!q.empty())
{
int top=q.front();
q.pop();
if(top==t)
{
cout<<dis[t]+1<<endl;
return ;
}
for(int i=1; i<=n; i++)
{
if(a[top][i]&&!visit[i])
{
visit[i]=1;
dis[i]=dis[top]+1;
q.push(i);
//cout<<"i="<<i<<" "<<dis[i]<<endl;
}
}
}
cout<<"-1"<<endl;
}
int main()
{
while(cin>>n>>m>>k)
{
memset(a,0,sizeof(a));
for(int i=0; i<m; i++)
{
int u,v;
int kk;
cin>>u>>kk;
while(kk--)
{
cin>>v;
a[u][v]=1;
}
}
while(k--)
{
memset(visit,0,sizeof(visit));
memset(dis,0,sizeof(dis));
cin>>t;
bfs(1);
}
}
}
求大佬指點哪裡錯了。。。
#include <bits/stdc++.h>
using namespace std;
vector<int>v[500];
int dis[500];
int visit[501];
int t;
void bfs(int x)
{
//cout<<t<<endl;
visit[x]=1;
dis[x]=1;
queue<int>q;
q.push(x);
while(!q.empty())
{
int top=q.front();
q.pop();
//cout<<top<<endl;
if(top==t)
{
//cout<<endl;
cout<<dis[t]<<endl;
return ;
}
for(int i=0; i<v[top].size(); i++)
{
if(!visit[v[top][i]])
{
visit[v[top][i]]=1;
dis[v[top][i]]=dis[top]+1;
//cout<<"i="<<v[top][i]<<" "<<dis[v[top][i]]<<endl;
q.push(v[top][i]);
}
}
}
cout<<"-1"<<endl;
}
int main()
{
int n,m,q;
while(cin>>n>>m>>q)
{
for(int i=0; i<m; i++)
{
int k,kk;
cin>>k>>kk;
while(kk--)
{
int k2;
cin>>k2;
v[k].push_back(k2);
}
}
while(q--)
{
memset(visit,0,sizeof(visit));//遍曆标記
memset(dis,0,sizeof(dis));//距離(lí)記錄
cin>>t;//要到的點
bfs(1);//從第一個(gè)點開始遍曆
}
v[500].clear();//清空
}
return 0;
}