[JAVA/코딩테스트] 송아지 찾기 (BFS : 상태트리탐색)

설명

현수는 송아지를 잃어버렸다. 다행히 송아지에는 위치추적기가 달려 있다.

현수의 위치와 송아지의 위치가 수직선상의 좌표 점으로 주어지면 현수는 현재 위치에서 송아지의 위치까지 다음과 같은 방법으로 이동한다.

송아지는 움직이지 않고 제자리에 있다.

현수는 스카이 콩콩을 타고 가는데 한 번의 점프로 앞으로 1, 뒤로 1, 앞으로 5를 이동할 수 있다.

최소 몇 번의 점프로 현수가 송아지의 위치까지 갈 수 있는지 구하는 프로그램을 작성하세요.

입력

첫 번째 줄에 현수의 위치 S와 송아지의 위치 E가 주어진다. 직선의 좌표 점은 1부터 10,000까지이다.

출력

점프의 최소횟수를 구한다. 답은 1이상이며 반드시 존재합니다.

예시 입력 1 

5 14

예시 출력 1

3

 

 

- 2022.01.25 풀이

import java.util.*;

public class Main {

    public static void solution(int S, int E) {

        // 결과값
        int result = 1;

        Queue<Integer> Q = new LinkedList<>();
        // 중복방지를 위한 arrayList
        ArrayList<Integer> arrayList = new ArrayList();

        Q.offer(S);

        while(!Q.isEmpty()){

            int len = Q.size();

            for(int i=0; i<len; i++){

                int compare = Q.poll();

                if(compare+1 == E || compare+5 == E || compare-1 == E){
                    System.out.println(result);
                    return;
                }
                
                // 중복방지
                if(!arrayList.contains(compare+1)){
                    Q.offer(compare+1);
                    arrayList.add(compare+1);
                }
                if(!arrayList.contains(compare+5)){
                    Q.offer(compare+5);
                    arrayList.add(compare+5);
                }
                if(!arrayList.contains(compare-1)){
                    Q.offer(compare-1);
                    arrayList.add(compare-1);
                }
            }
            result++;
        }

    }
    
    public static void main(String[] args) {

        Main main = new Main();
        Scanner kb = new Scanner(System.in);

        // 현수의 위치
        int S = kb.nextInt();
        // 송아지의 위치
        int E = kb.nextInt();

        main.solution(S, E);
    }
}

- 코드

import java.util.*;

class Main{
	int answer = 0;
	int[] dis= {1, -1, 5};
	// 한번 방문 했던것을 저장 - 시간복잡도 줄이기 위함
	int[] ch;
	Queue<Integer> Q = new LinkedList<>();
	public int BFS(int s, int e) {
		
		ch = new int[10001];
		ch[s] = 1;
		Q.offer(s);
		int Level = 0;
		
		while(!Q.isEmpty()) {
			int len=Q.size();
			
			for(int i=0; i<len; i++) {
				int x = Q.poll();
				// 방법 1
				//if(x==e) {
				//	return Level;
				//}	
				for(int j=0; j<3; j++) {
						int nx = x+dis[j];
						// 방법 2
						if(nx==e) {
							return Level + 1;
						}
						// 방문 안한거
						if(nx >=1 && nx<=10000 && ch[nx] == 0) {
							ch[nx] = 1;
							Q.offer(nx);
						}
					}	
				
			}
			Level ++;
		}
		
		return Level;
	}
	
	public static void main(String[] args) {
		Main T = new Main();
		Scanner in = new Scanner(System.in);
		int s = in.nextInt();
		int e = in.nextInt();
		System.out.println(T.BFS(s, e));
	}
	
}