문제풀이/백준oj
[백준OJ] 5014번 스타트링크
Hyeon-Uk
2021. 7. 22. 20:33
반응형
https://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를 이용하여, 모든 경우중 최소횟수로 갈 수 있는 경우들을 구한 뒤, 만약 G층에 도달한다면 그 횟수를, 도달하지 못했다면 "use the stairs"를 출력해준다.
-시간복잡도-
모든 층을 검사하므로, O(F)가 된다.
-코드-
#include <iostream>
#include<algorithm>
#include<queue>
using namespace std;
int F, S, G, U, D;
void updown() {
bool visited[1000001] = { 0 };
queue<pair<int, int>> q;
q.push({ S,0 });
while (!q.empty()) {
int now = q.front().first;
int cnt = q.front().second;
q.pop();
if (now == G) {
cout << cnt << "\n";
return;
}
if (now + U <= F && !visited[now + U]) {
q.push({ now + U,cnt + 1 });
visited[now + U] = true;
}
if (now - D >= 1 && !visited[now - D]) {
q.push({ now - D,cnt + 1 });
visited[now - D] = true;
}
}
cout << "use the stairs\n";
return;
}
int main(){
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
cin >> F >> S >> G >> U >> D;
updown();
return 0;
}
반응형