반응형
문제
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 구조를 선언하여 가능한 경우를 저장했습니다.
반응형
'코딩테스트 > DFS & BFS' 카테고리의 다른 글
[Java] 백준 10026번 : 적록색약 (0) | 2022.02.23 |
---|---|
[Java] 백준 1260번 : DFS와 BFS (0) | 2022.02.21 |
[Java] 백준 1743번 : 음식물 피하기 (0) | 2022.02.15 |
[Python] 백준 1967번 : 트리의 지름 (0) | 2022.01.07 |
[Python] 백준 1167번 : 트리의 지름 (0) | 2022.01.06 |