[백준] 1913번 - 달팽이 (Silver 3)

문제 링크

[백준] 1913번 - 달팽이 (Silver 3)

문제 설명

홀수인 자연수 N이 주어져있을 때, N x N 표에 달팽이 모양으로 숫자를 채워서 출력한다.
$N^2$ 이하의 자연수의 좌표도 함께 출력한다.

N = 5일 때 달팽이 모양으로 숫자를 채우는 예시는 아래와 같다.

25	10	11	12	13
24	9	2	3	14
23	8	1	4	15
22	7	6	5	16
21	20	19	18	17

정답 코드 및 설명

달팽이 모양으로 숫자를 채우는 규칙은 아래와 같이 요약할 수 있다.

1. $1^2$을 가운데에 넣고 위로 이동한다.
2. $2=1^2+1$을 넣고 오른쪽으로 이동해서 3을 넣는다.
3. 아래로 이동해서 4를 넣고, 또 아래로 이동해서 5를 넣는다.
4. 왼쪽으로 이동해서 6을 넣고, 또 왼쪽으로 이동해서 7을 넣는다.
5. 위로 위동해서 8을 넣고, 또 위로 이동해서 $9=3^2$을 넣고 위로 이동한다.
6. $10=3^2+1$을 넣고 오른쪽으로 이동해서 11, 12, 13을 넣는다.
7. 비슷하게 계속 반복한다.

이를 구현한 코드가 아래에 있다.

import java.io.*;

public class BOJ1913 {
    int n, num, numX, numY; // num의 좌표 = (numX, numY)
    int[][] table;

    void input() throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        n = Integer.parseInt(br.readLine());
        num = Integer.parseInt(br.readLine());
        table = new int[n][n];
    }

    void makeTable() {
        int x = (n - 1) / 2, y = (n - 1) / 2;
        if (num == 1) { // num이 1인 경우의 좌표
            numX = x;
            numY = y;
        }
        table[x--][y] = 1; // 위로 이동
        for (int i = 1; i < n; i += 2) {
            for (int j = i * i + 1; j <= (i + 2) * (i + 2); j++) {
                int k = (j - i * i) / (i + 1); // 어느 방향으로 이동해야하는지 판별
                if (num == j) { // num이 j인 경우의 좌표
                    numX = x;
                    numY = y;
                }
                switch (k) {
                    case 0 -> table[x][y++] = j; // 오른쪽으로 이동
                    case 1 -> table[x++][y] = j; // 아래로 이동
                    case 2 -> table[x][y--] = j; // 왼쪽으로 이동
                    default -> table[x--][y] = j; // 위로 이동
                }
            }
        }
    }

    void print() throws IOException {
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++)
                sb.append(table[i][j]).append(' ');
            sb.append('\n');
        }
        sb.append(numX + 1).append(' ').append(numY + 1);
        bw.write(sb.toString());
        bw.close();
    }

    void solution() throws IOException {
        input();
        makeTable();
        print();
    }

    public static void main(String[] args) throws IOException {
        new BOJ1913().solution();
    }
}