[백준] CLASS3 1074 Z - JAVA [자바]

안녕하세요. 이번에는 백준 1074 Z  문제를 풀어보려고 합니다.

 

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

1074번: Z

---

Problem

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

 

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

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

 

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

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

사진 1. 문제

---

Solution

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

public class Main {
	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());
		int N = Integer.parseInt(st.nextToken());
		int r = Integer.parseInt(st.nextToken());
		int c = Integer.parseInt(st.nextToken());
		int size = (int) Math.pow(2, N);
		
		find(size, r, c);
		System.out.println(count);
	}

	private static void find(int size, int r, int c) {
		if(size == 1)
			return;
		
        int newSize = size / 2;
        // 왼쪽 위
		if(r < newSize && c < newSize) {
			find(newSize, r, c);
		}
        // 오른쪽 위
		else if(r < newSize && c >= newSize) {
			count += size * size / 4;
			find(newSize, r, c - newSize);
		}
        // 왼쪽 아래
		else if(r >= newSize && c < newSize) {
			count += (size * size / 4) * 2;
			find(newSize, r - newSize, c);
		}
        // 오른쪽 아래
		else {
			count += (size * size / 4) * 3;
			find(size / 2, r - newSize, c - newSize);
		}
	}
}

---

Performance

사진 2. 실행 결과