알고리즘/L20 알고리즘
20. 최단거리
쿠쿠트레인
2023. 7. 19. 15:12
문제
그래프와 출발점, 도착점이 주어질 때 출발점에서 도착점까지 이동하기 위한 최단거리를 출력하는 프로그램을 작성하시오. 예를 들어, 아래 그림에서 출발 정점이 0, 도착 정점이 10이라고 할 때, 최단거리는 3이다.
입력
첫째 줄에 정점의 개수 N과 간선의 개수 M이 주어진다. ( 1 ≤ N ≤ 10,000, 1 ≤ M ≤ 1,000,000 ) 둘째 줄부터 간선의 정보가 주어진다. 각 줄은 두 개의 숫자 a, b로 이루어져 있으며, 이는 정점 a와 정점 b가 연결되어 있다는 의미이다. M+1 번째 줄에 대하여 출발점과 도착점의 정점 번호가 주어진다. 정점의 번호는 0번부터 N-1번까지이다.
출력
출발점에서 도착점까지 이동하기 위한 최단거리를 출력한다.
예제 입력
11 14
0 1
0 2
1 2
1 4
1 5
2 3
3 7
4 7
4 9
4 10
5 6
6 8
6 10
7 8
0 10
예제 출력
3
#include <cstdio>
#include <vector>
#include <queue>
using namespace std;
const int MAX = 1000005;
const int nMAX = 10005;
vector <int> graph[MAX];
int arr[nMAX] = {0,};
queue <int> q;
int n,m;
void BFS(){
if(q.empty()) return;
int node = q.front();
q.pop();
for(int i = 0;i <graph[node].size();i++){
if(arr[graph[node][i]] == 0){
q.push(graph[node][i]);
arr[graph[node][i]] = arr[node] + 1;
}
}
if(!q.empty()) BFS();
}
int main() {
scanf("%d %d",&n,&m);
for(int i=0;i<m;i++){
int a,b;
scanf("%d %d",&a,&b);
graph[a].push_back(b);
graph[b].push_back(a);
}
int start,end2;
scanf("%d %d",&start,&end2);
q.push(start);
arr[start] = 0;
BFS();
printf("%d",arr[end2]);
return 0;
}