백준 1074번-Z(JAVA)

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일 때의 예이다.

입력

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

출력

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

 

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

public class z {
	static int p = 0;
	static int r;
	static int c;

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine(), " ");
		int N = Integer.parseInt(st.nextToken());
		r = Integer.parseInt(st.nextToken());
		c = Integer.parseInt(st.nextToken());
		visit(0, 0, N);
	}

	static void visit(int row, int col, int cnt) {
		if (row == r && col == c) {
			System.out.println(p);
			return;
		}
		int t = (int) Math.pow(2, cnt - 1);
		if (r < row + t && c < col + t) {
			visit(row, col, cnt - 1);
			return;
		}
		if (r < row + t && c >= col + t) {
			p += t * t * 1;
			visit(row, col + t, cnt - 1);
			return;
		}
		if (r >= row + t && c < col + t) {
			p += t * t * 2;
			visit(row + t, col, cnt - 1);
			return;
		}
		if (r >= row + t && c >= col + t) {
			p += t * t * 3;
			visit(row + t, col + t, cnt - 1);
			return;
		}
	}
}

 

r과 c의 값에 따라 4분면을 지정해 탐색하도록 풀이해 주었습니다.