코딩테스트/DFS & BFS

[Java] 백준 2210번 : 숫자판 점프

sujin7837 2022. 2. 15. 21:28
반응형

문제

5×5 크기의 숫자판이 있다. 각각의 칸에는 숫자(digit, 0부터 9까지)가 적혀 있다. 이 숫자판의 임의의 위치에서 시작해서, 인접해 있는 네 방향으로 다섯 번 이동하면서, 각 칸에 적혀있는 숫자를 차례로 붙이면 6자리의 수가 된다. 이동을 할 때에는 한 번 거쳤던 칸을 다시 거쳐도 되며, 0으로 시작하는 000123과 같은 수로 만들 수 있다.

숫자판이 주어졌을 때, 만들 수 있는 서로 다른 여섯 자리의 수들의 개수를 구하는 프로그램을 작성하시오.

입력

다섯 개의 줄에 다섯 개의 정수로 숫자판이 주어진다.

출력

첫째 줄에 만들 수 있는 수들의 개수를 출력한다.

예제 입력 1

1 1 1 1 1
1 1 1 1 1
1 1 1 1 1
1 1 1 2 1
1 1 1 1 1

예제 출력 1

15

힌트

111111, 111112, 111121, 111211, 111212, 112111, 112121, 121111, 121112, 121211, 121212, 211111, 211121, 212111, 212121 이 가능한 경우들이다.

 

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.StringTokenizer;

public class Main {

	private static int[][] map=new int[5][5];
	private static HashSet<Integer> set=new HashSet<Integer>();
	private static boolean [][] visited=new boolean[5][5];
	private static int [] dx = {-1,1,0,0};
	private static int [] dy = {0,0,-1,1};
	private static int result=0;
	private static StringTokenizer st;
	
	public static void main(String[] args) throws IOException {

		BufferedReader bf=new BufferedReader(new InputStreamReader(System.in));
		
		for(int r=0;r<5;r++) {
			st=new StringTokenizer(bf.readLine());
			for(int c=0;c<5;c++) {
				map[r][c]=Integer.parseInt(st.nextToken());
			}
		}
		for(int r=0;r<5;r++) {
			for(int c=0;c<5;c++) {
				dfs(r, c, 0, 0);
			}
		}
		System.out.println(set.size());
	}

	public static void dfs(int x, int y, int cnt, int result) {
		if(cnt==6) {
			set.add(result);
			return;
		}
		
		for(int i=0;i<4;i++) {
			int nx=x+dx[i];
			int ny=y+dy[i];
			
			if(isIn(nx, ny)) dfs(nx, ny, cnt+1, result*10+map[nx][ny]);
		}
	}
	
	public static boolean isIn(int x, int y) {
		return x>=0 && x<5 && y>=0 && y<5;
	}
}

중복되는 경우를 제외해주기 위해 set 구조를 선언하여 가능한 경우를 저장했습니다.

반응형