[백준] 1074 Z Java

문제

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

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

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

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

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

 

입력

첫째 줄에 정수 N, r, c가 주어진다.

 

출력

r행 c열을 몇 번째로 방문했는지 출력한다.

 

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main {
    static int N;
    static int r;
    static int c;
    static int count = 0;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        N = Integer.parseInt(st.nextToken());
        r = Integer.parseInt(st.nextToken());
        c = Integer.parseInt(st.nextToken());
        int size = (int) Math.pow(2, N);   //배열 크기

        search(size, r, c);
        System.out.println(count);
    }

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

문제에 재귀적으로 라는 말이 있어서 재귀 사용해서 푸는 거구나 감 잡고 시작했다.

r과 c에 따라 어느 사분면에 위치하는지 나뉘기 때문에 if문으로 나눠줬고, 2, 3, 4사분면에 위치한 경우 그 전 카운트를 더해주고 재귀함수를 호출했다.

 

문제 출처

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