[백준] 16926번 배열 돌리기 1 (자바 풀이)

문제

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

16926번: 배열 돌리기 1

 

풀이

이 문제는 단순 구현 문제이지만, 풀기 쉽지 않았다. 내 풀이는 다음과 같다.

[그림 1] 예제1

[그림 1]과 같이 temp 배열을 만들고, 사각형의 테두리를 기준으로 왼쪽 위부터 반시계방향으로 원소를 탐색하면서 R칸씩 밀어서 (범위를 벗어나면 나머지 연산) 저장해놓는다. 

 

이후 다시 똑같은 순서로 탐색하면서 temp 배열에 저장해놓은 값을 붙여넣는다.

 

[그림2] 예제2

그 다음 사각형 테두리를 탐색하면서 똑같은 과정을 반복해준다.

 

코드

import java.io.*;
import java.util.*;
 
public class Baekjoon_16926 {
    public static void main(String[] args)throws Exception {
        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];
        
        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());
            }
        }
        
        int x = 0;
        int y = 0;
        int n = N;
        int m = M;
        while(x<N/2 && y<M/2) {
            int len = n*2+m*2-4;
            int idx = 0;
            int temp[] = new int[len];
            
            // 왼쪽
            for(int i=0; i<n; i++) {
                temp[(idx+R)%len] = arr[x+i][y];
                idx++;
            }
            idx--;
            // 아래
            for(int i=0; i<m; i++) {
                temp[(idx+R)%len] = arr[N-1-x][y+i];
                idx++;
            }
            idx--;
            // 오른쪽
            for(int i=0; i<n; i++) {
                temp[(idx+R)%len] = arr[N-x-i-1][M-1-y];
                idx++;
            }
            idx--;
            // 위
            for(int i=0; i<m; i++) {
                temp[(idx+R)%len] = arr[x][M-y-i-1];
                idx++;
            }
            
            idx = 0; 
            // 왼쪽
            for(int i=0; i<n; i++) {
                arr[x+i][y] = temp[idx++];
            }
            idx--;
            // 아래
            for(int i=0; i<m; i++) {
                arr[N-1-x][y+i]= temp[idx++];
            }
            idx--;
            // 오른쪽
            for(int i=0; i<n; i++) {
                arr[N-x-i-1][M-1-y] = temp[idx++];
            }
            idx--;
            // 위
            for(int i=0; i<m-1; i++) {
                arr[x][M-y-i-1] = temp[idx++];
            }            
            x++;
            y++;
            n -= 2;
            m -= 2;
            if(x==N/2 && y==M/2) {
                if(N%2==1 && M%2 ==1) {
                    arr[x][y] = arr[x][y];
                }
                break;
            }
        }
        
        StringBuilder sb = new StringBuilder();
        
        for(int i=0; i<N; i++) {
            for(int j=0; j<M; j++) {
                sb.append(arr[i][j] +" ");
            }
            sb.append("\n");
        }
        
        System.out.println(sb.toString());
    }
 
}
 
 

 

결과

[그림 3] 결과

쉽지 않았던 문제였다. 더 나은 풀이법이 존재할 것 같다.