[백준 알고리즘]/[C++]

5014번 스타트링크 | BFS | 백준 5014 C++ 코드

말하는펭귄 2021. 3. 3. 11:18
728x90
반응형

 

 

이번 포스팅은 백준 5014번 스타트링크입니다.

아래 url를 클릭하시면 백준 사이트에서 문제를 볼 수 있습니다.

www.acmicpc.net/problem/5014

 

5014번: 스타트링크

첫째 줄에 F, S, G, U, D가 주어진다. (1 ≤ S, G ≤ F ≤ 1000000, 0 ≤ U, D ≤ 1000000) 건물은 1층부터 시작하고, 가장 높은 층은 F층이다.

www.acmicpc.net

 

 

 

기본 알고리즘

최단 거리 = BFS (너비 우선 탐색)

 

 

 

전체 코드

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
//백준5014 스타트링크                                                        
 
#include <iostream>
#include <queue>
#include <vector>
using namespace std;
 
int F, S, G, U, D;
const int MAX = 1000001;
int path[MAX];
bool visited[MAX];
vector<int> dx;
queue<int> q;
 
void BFS(int v) {
    visited[v] = true;
    q.push(v);
 
    while (!q.empty()) {
        v = q.front();
        q.pop(); 
        
        for (int i = 0; i < 2; i++) {
            int nv = v + dx[i];
 
            if (0 < nv && nv <= F) {
                if (!visited[nv]) {
                    visited[nv] = true;
                    q.push(nv);
                    path[nv] = path[v] + 1;
                }
            }
        }
        
    }
}
 
int main() {
    cin >> F >> S >> G >> U >> D;
 
    dx.push_back(U);
    dx.push_back(D * -1);
 
    BFS(S);
 
    if (visited[G]) {
        cout << path[G];
    }
    else {
        cout << "use the stairs";
    }
}
cs

 

 

 

 

728x90
반응형