[BOJ][Java] 16926. 배열 돌리기 1

문제

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

 

문제 이해

• 주어진 n x m 크기의 2차원 배열을 반시계 방향으로 r번 회전시킨다.
• 배열은 여러 겹의 레이어(테두리)로 구성되어 있으며, 각 회전마다 바깥쪽 레이어부터 안쪽 레이어까지 순차적으로 회전시킨다.

 

문제 풀이

• 배열의 회전 가능한 레이어 수: 배열의 가장 짧은 변의 절반만큼 레이어가 존재한다.
  line = Math.min(n, m) / 2
• r번 회전할 동안, 각 레이어에 대해 4방향 이동을 반복 수행
• 회전할 때는 현재 위치에서 다음 위치로 값을 덮어쓰기 하므로 맨 처음 값을 임시 저장해두고 마지막에 채워 넣는다.

 

회전 과정

• 각 레이어에 대해 다음과 같은 방향으로 요소를 한 칸씩 이동
  • → 오른쪽: 좌측 열 → 우측 열 (0, +1)
  • ↓ 아래쪽: 상단 행 → 하단 행 (+1, 0)
  • ← 왼쪽: 우측 열 → 좌측 열 (0, -1)
  • ↑ 위쪽: 하단 행 → 상단 행 (-1, 0)
• 방향 전환은 범위를 벗어나면 방향 인덱스를 증가시키는 방식으로 처리

 

회전 범위

레이어 i에 대해 이동 가능한 범위

• 행(row): i부터 n - i - 1까지
• 열(col): i부터 m - i - 1까지

레이어 i 구성

(i, i)                →       (i, m - i - 1)
  ↓                                 ↓
(n - i - 1, i)    ←       (n - i - 1, m - i - 1)

 

코드

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

public class Main {
	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 m = Integer.parseInt(st.nextToken());
		int r = Integer.parseInt(st.nextToken());
		int[][] arr = new int[n][m];
		int layer = Math.min(n, m) / 2; // 회전 레이어 수
		// 왼쪽으로, 위로, 오른쪽으로, 아래로
		int[] dr = { 0, 1, 0, -1 };
		int[] dc = { 1, 0, -1, 0 };

		for (int i = 0; i < n; i++) {
			st = new StringTokenizer(br.readLine());
			for (int j = 0; j < m; j++) {
				arr[i][j] = Integer.parseInt(st.nextToken());
			}
		}

		// 연산
		while (r-- > 0) {
			for (int i = 0; i < layer; i++) {
				int idx = 0;	// 방향
				int row = i, col = i;
				int temp = arr[i][i]; // 첫 번째 값 저장해놓기
				while (idx < 4) {
					int nr = row + dr[idx];
					int nc = col + dc[idx];

					if (nr < i || nc < i || nr >= n - i || nc >= m - i) {	// 범위 벗어나면
						idx++;
						continue;
					}

					arr[row][col] = arr[nr][nc];
					row = nr;
					col = nc;
				}
				arr[i + 1][i] = temp;
			}
		}

		// 출력
		StringBuilder sb = new StringBuilder();
		for (int[] rows : arr) {
			for (int i : rows) {
				sb.append(i).append(" ");
			}
			sb.append("\n");
		}
		System.out.println(sb);
	}
}