[백준] 2239: 스도쿠 (Java)

Yoon Uk·2022년 7월 29일
0

알고리즘 - 백준

목록 보기
41/130
post-thumbnail
post-custom-banner

문제

BOJ 2239: 스도쿠 https://www.acmicpc.net/problem/2239

풀이

조건

  • 일반 스도쿠의 조건과 같다.
  • 1~9까지의 수가 가로, 세로, 9개의 네모 칸 안에서 중복되게 있으면 안된다.
  • 답이 여러 개 있다면 그 중 사전식으로 앞서는 것을 출력한다. 즉, 81자리의 수가 제일 작은 경우를 출력한다.

풀이 순서

  • 입력 받을 때 빈 칸의 좌표list에 따로 저장해둔다.
  • 재귀 호출을 할 때 마다 해당 재귀의 깊이list의 크기를 비교하며 같아지면 종료한다.
    • 이 때 가능한 조합이 만들어지면 출력하고 즉시 종료한다.
  • 가로, 세로, 네모 칸 안을 비교하며 조건에 맞으면 재귀 호출을 한다.

코드

import java.util.*;
import java.io.*;

public class Main {
	
	static final int N = 9;
	static int[][] map = new int[N][N];
	static ArrayList<int[]> list = new ArrayList<>(); // 빈 칸의 좌표가 들어갈 리스트
	
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		for(int i=0; i<N; i++) {
			String str = br.readLine();
			for(int j=0; j<N; j++) {
				map[i][j] = str.charAt(j) - '0';
				// 빈 칸의 좌표를 리스트에 넣음
				if(map[i][j] == 0) {
					list.add(new int[] {i, j});
				}
			}
		}
		
		bt(0);
		
	}
	
	static void bt(int depth) {
		
		if(list.size() == depth) {
			for(int i=0; i<9; i++) {
				for(int j=0; j<9; j++) {
					System.out.print(map[i][j]);
				}
				System.out.println();
			}
			
			// 여러 케이스가 나올 수 있으므로 return 대신 
			// 시스템 종료를 통해 최초로 조합된 경우만 출력하고 끝낸다.
			System.exit(0);
		}
		
		
		int x = list.get(depth)[0];
		int y = list.get(depth)[1];
		
		// 이미 있는 숫자인지 체크할 배열
		boolean[] check = new boolean[10];
		
		// 가로 체크
		for(int i=0; i<9; i++) {
			if(map[x][i] != 0) {
				check[map[x][i]] = true;
			}
		}
		
		// 세로 체크
		for(int i=0; i<9; i++) {
			if(map[i][y] != 0) {
				check[map[i][y]] = true;
			}
		}
		
		// 네모 체크
		int firstX = (x / 3) * 3;
		int firstY = (y / 3) * 3;
		for(int i = firstX; i < firstX + 3; i++) {
			for(int j = firstY; j < firstY + 3; j++) {
				if(map[i][j] != 0) {
					check[map[i][j]] = true;
				}
			}
		}
		
		// 넣을 수 있는 숫자 넣음
		for(int i=1; i<=9; i++) {
			if(!check[i]) {
				map[x][y] = i;
				bt(depth + 1);
				map[x][y] = 0;
			}
		}
		
	}	
	
}

정리

  • 처음에 재귀 호출 시 체크해야 할 조건을 복잡하게 구현해서 다시 풀었다.
post-custom-banner

0개의 댓글