[Baekjoon] Class3 03 - Z[1074]

Z [1074번]

한수는 크기가 2N × 2N인 2차원 배열을 Z모양으로 탐색하려고 한다. 예를 들어, 2×2배열을 왼쪽 위칸, 오른쪽 위칸, 왼쪽 아래칸, 오른쪽 아래칸 순서대로 방문하면 Z모양이다.
N > 1인 경우, 배열을 크기가 $2^N-1$ × $2^N-1$로 4등분 한 후에 재귀적으로 순서대로 방문한다.
다음 예는 22 × 22 크기의 배열을 방문한 순서이다.

N이 주어졌을 때, r행 c열을 몇 번째로 방문하는지 출력하는 프로그램을 작성하시오.
다음은 N=3일 때의 예이다.

입력

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

출력

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

생각해볼점

공간을 4분할하며 재귀를 사용한다.
확인할 장소의 위치가 현재 공간의 몇 사분면인지 파악한 후, 해당 공간으로 들어가 이를 반복하는 방식으로 진행하였다.

코드 구현

#include <stdio.h>
#include <stdlib.h>
#include <math.h>

static void z_recur(int size, int row, int column, int row_idx, int column_idx, int *ans)
{
	if (size < 2)  // 공간의 가로세로가 2보다 작다면 끝
		return ;
	if (row < row_idx - size / 2 && column < column_idx - size / 2)  // 2사분면
		z_recur(size / 2, row, column, row_idx - size / 2, column_idx - size / 2, ans);
	else if (row < row_idx - size / 2 && column >= column_idx - size / 2)  // 1사분면
	{
		*ans += pow(size / 2, 2);
		z_recur(size / 2, row, column, row_idx - size / 2, column_idx, ans);
	}
	else if (row >= row_idx - size / 2 && column < column_idx - size / 2)  // 3사분면
	{
		*ans += pow(size / 2, 2) * 2;
		z_recur(size / 2, row, column, row_idx, column_idx - size / 2, ans);
	}
	else if (row >= row_idx - size / 2 && column >= column_idx - size / 2)  // 4사분면
	{
		*ans += pow(size / 2, 2) * 3;
		z_recur(size / 2, row, column, row_idx, column_idx, ans);
	}
}

int main()
{
	int	num, size, row, column, ans = 0;

	scanf("%d %d %d", &num, &row, &column);
	size = pow(2, num);

	z_recur(size, row, column, size, size, &ans);

	printf("%d", ans);
	return (0);
}