본문 바로가기
백준

[백준] C++ 1074번 Z

by 당코 2023. 2. 3.

문제

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

 

1074번: Z

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

www.acmicpc.net

 

접근 방법

주어진 배열을 좌표라 생각을 하면 4개의 사분면으로 나눌 수 있다.

어느 사분면에 따라 앞서 방문해야 할 칸의 개수가 달라진다.

좌표가 3사분면에 있다고 가정할 때, 1사분면과 2사분면의 모든 칸들을 방문하여야 한다.

앞서 방문해야 할 칸들의 값을 모두 더하고, 현재 좌표가 위치한 사분면을 기준으로 다시 4등분을 하여 n=1이 될 때까지 반복으로 수행하였다.

 

코드

#include <iostream>
#include <cmath>
using namespace std;

int n, r, c, ans;

int main() {
	cin >> n >> r >> c;
	while (n > 1) {
		int size = pow(2, n);
		int partSize = pow(size / 2, 2);
		if (r < size / 2 && c >= size / 2) {
			ans += partSize;
			c -= size / 2;
		}
		else if (r >= size / 2 && c < size / 2) {
			ans += 2 * partSize;
			r -= size / 2;
		}
		else if (r >= size / 2 && c >= size / 2) {
			ans += 3 * partSize;
			c -= size / 2;
			r -= size / 2;
		n--;
		}
	}
	cout << ans + 2 * r + c << '\n';
}

마지막에 n=1 이 되어 칸이 총 4개만 남은 경우에 좌표에 따라 0,1,2,3을 더해주었다.