백준 1074 자바 - Z

https://www.acmicpc.net/problem/1074

1074번: Z

문제

한수는 크기가 2N × 2N인 2차원 배열을 Z모양으로 탐색하려고 한다. 예를 들어, 2×2배열을 왼쪽 위칸, 오른쪽 위칸, 왼쪽 아래칸, 오른쪽 아래칸 순서대로 방문하면 Z모양이다.

N > 1인 경우, 배열을 크기가 2N-1 × 2N-1로 4등분 한 후에 재귀적으로 순서대로 방문한다.

다음 예는 22 × 22 크기의 배열을 방문한 순서이다.

N이 주어졌을 때, r행 c열을 몇 번째로 방문하는지 출력하는 프로그램을 작성하시오.

다음은 N=3일 때의 예이다.

코드

import java.io.*;

public class Main {

    private static int count;

    public static void main(String[] args) throws IOException {
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        String[] input = br.readLine().split(" ");
        int n = Integer.parseInt(input[0]), r = Integer.parseInt(input[1]),
                c = Integer.parseInt(input[2]), size = (int) Math.pow(2, n);

        find(size, r, c);

        bw.write(String.valueOf(count));
        bw.flush();
        bw.close();
        br.close();

    }

    private static void find(int size, int r, int c) {
        if (size == 1) {
            return;
        }
        if (r < size / 2 && c < size / 2) {
            find(size / 2, r, c);
        }
        else if (r < size / 2 && c >= size / 2) {
            count += size * size / 4;
            find(size / 2, r, c - size / 2);
        }
        else if (r >= size / 2 && c < size / 2) {
            count += (size * size / 4) * 2;
            find(size / 2, r - size / 2, c);
        }
        else {
            count += (size * size / 4) * 3;
            find(size / 2, r - size / 2, c - size / 2);
        }
    }

}

풀이

재귀함수가 있는데, 4분면의 각 부분들에 대한 연산을 if문으로 걸러서 수행하면 된다. size 가 1이 되면 멈춰서 count에 더해줬던 값을 출력한다.