백준 12761번. 돌다리 (c++) / BFS

dist배열에 8개 원소를 넣었다. 

0~3 => 오른쪽으로 1, -1, a, b 

4~5 => 왼쪽으로 a, b

6~7 => 힘 모아서 오른쪽으로 a, b배

 

이후에는, bfs로 최단 경로를 찾아주었다.

 

#include <iostream>
#include <queue>
#include <cstdio>
 
using namespace std;
 
int a, b, n, m;
bool visited[100001] = {false, };
int stone[100001] = {0, };
 
int main()
{
    scanf("%d %d %d %d", &a, &b, &n, &m);    
    
    const int dist[8] = {1, -1, a, b, -a, -b, a, b};
    queue<int> q;
    visited[n] = true;
    stone[n] = 0;
    q.push(n);
 
    while(!q.empty())
    {
        int curr = q.front();
        q.pop();
 
        if(curr == m)
        {
            printf("%d", stone[m]);
            break;
        }
 
        // 일반 점프
        for(int i=0; i<6; i++)
        {
            int next = curr+dist[i];
 
            if(next<0 || 100000<next) continue;
 
            if(!visited[next])
            {
                visited[next] = true;
                stone[next] = stone[curr]+1;
                q.push(next);
            }
        }
 
 
        // 힘 모아서 점프
        for(int i=6; i<8; i++)
        {
            int next = curr*dist[i];
            
            if(next<0 || 100000<next) continue;
 
            if(!visited[next])
            {
                visited[next] = true;
                stone[next] = stone[curr]+1;
                q.push(next);
            }
        }
 
    }
 
    return 0;
}