백준 16935 (배열 돌리기 3) - java

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

16935번: 배열 돌리기 3

문제

크기가 N×M인 배열이 있을 때, 배열에 연산을 R번 적용하려고 한다. 연산은 총 6가지가 있다.

1번 연산은 배열을 상하 반전시키는 연산이다.

1 6 2 9 8 4 → 4 2 9 3 1 8
7 2 6 9 8 2 → 9 2 3 6 1 5
1 8 3 4 2 9 → 7 4 6 2 3 1
7 4 6 2 3 1 → 1 8 3 4 2 9
9 2 3 6 1 5 → 7 2 6 9 8 2
4 2 9 3 1 8 → 1 6 2 9 8 4
   <배열>       <연산 결과>

2번 연산은 배열을 좌우 반전시키는 연산이다.

1 6 2 9 8 4 → 4 8 9 2 6 1
7 2 6 9 8 2 → 2 8 9 6 2 7
1 8 3 4 2 9 → 9 2 4 3 8 1
7 4 6 2 3 1 → 1 3 2 6 4 7
9 2 3 6 1 5 → 5 1 6 3 2 9
4 2 9 3 1 8 → 8 1 3 9 2 4
   <배열>       <연산 결과>

3번 연산은 오른쪽으로 90도 회전시키는 연산이다.

1 6 2 9 8 4 → 4 9 7 1 7 1
7 2 6 9 8 2 → 2 2 4 8 2 6
1 8 3 4 2 9 → 9 3 6 3 6 2
7 4 6 2 3 1 → 3 6 2 4 9 9
9 2 3 6 1 5 → 1 1 3 2 8 8
4 2 9 3 1 8 → 8 5 1 9 2 4
   <배열>       <연산 결과>

4번 연산은 왼쪽으로 90도 회전시키는 연산이다.

1 6 2 9 8 4 → 4 2 9 1 5 8
7 2 6 9 8 2 → 8 8 2 3 1 1
1 8 3 4 2 9 → 9 9 4 2 6 3
7 4 6 2 3 1 → 2 6 3 6 3 9
9 2 3 6 1 5 → 6 2 8 4 2 2
4 2 9 3 1 8 → 1 7 1 7 9 4
   <배열>       <연산 결과>

5, 6번 연산을 수행하려면 배열을 크기가 N/2×M/2인 4개의 부분 배열로 나눠야 한다. 아래 그림은 크기가 6×8인 배열을 4개의 그룹으로 나눈 것이고, 1부터 4까지의 수로 나타냈다.

1 1 1 1 2 2 2 2
1 1 1 1 2 2 2 2
1 1 1 1 2 2 2 2
4 4 4 4 3 3 3 3
4 4 4 4 3 3 3 3
4 4 4 4 3 3 3 3

5번 연산은 1번 그룹의 부분 배열을 2번 그룹 위치로, 2번을 3번으로, 3번을 4번으로, 4번을 1번으로 이동시키는 연산이다.

3 2 6 3 1 2 9 7 → 2 1 3 8 3 2 6 3
9 7 8 2 1 4 5 3 → 1 3 2 8 9 7 8 2
5 9 2 1 9 6 1 8 → 4 5 1 9 5 9 2 1
2 1 3 8 6 3 9 2 → 6 3 9 2 1 2 9 7
1 3 2 8 7 9 2 1 → 7 9 2 1 1 4 5 3
4 5 1 9 8 2 1 3 → 8 2 1 3 9 6 1 8
     <배열>            <연산 결과>

6번 연산은 1번 그룹의 부분 배열을 4번 그룹 위치로, 4번을 3번으로, 3번을 2번으로, 2번을 1번으로 이동시키는 연산이다.

3 2 6 3 1 2 9 7 → 1 2 9 7 6 3 9 2
9 7 8 2 1 4 5 3 → 1 4 5 3 7 9 2 1
5 9 2 1 9 6 1 8 → 9 6 1 8 8 2 1 3
2 1 3 8 6 3 9 2 → 3 2 6 3 2 1 3 8
1 3 2 8 7 9 2 1 → 9 7 8 2 1 3 2 8
4 5 1 9 8 2 1 3 → 5 9 2 1 4 5 1 9
     <배열>            <연산 결과>

유형 : 구현

삼성에서 좋아하는 배열 조작 구현 문제이다. 기본 회전의 규칙을 찾는다면 쉽게 해결할 수 있다.

 

전체 코드

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

public class Main {
    static int n = 0;
    static int m = 0;
    static int k = 0;

    static int[][] board;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        n = Integer.parseInt(st.nextToken());
        m = Integer.parseInt(st.nextToken());
        k = Integer.parseInt(st.nextToken());

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

        st = new StringTokenizer(br.readLine(), " ");
        for(int i=1;i<=k;i++){
            int what = Integer.parseInt(st.nextToken());

            if(what == 1){
                first();
            }

            else if(what == 2){
                second();
            }

            else if(what == 3){
                three();
            }

            else if(what == 4){
                four();
            }

            else if(what == 5){
                five();
            }

            else{
                six();
            }
        }

        StringBuilder sb = new StringBuilder();
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                sb.append(board[i][j]+" ");
            }sb.append("\n");
        }

        System.out.print(sb);
        br.close();
    }

    static void first(){
        int[][] tmp = new int[n][m];

        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                tmp[i][j] = board[n-1-i][j];
            }
        }

        board = tmp;
    }

    static void second(){
        int[][] tmp = new int[n][m];

        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                tmp[i][j] = board[i][m-1-j];
            }
        }

        board = tmp;
    }

    static void three(){
        int[][] tmp = new int[m][n];

        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                tmp[j][n-1-i] = board[i][j];
            }
        }

        board = tmp;
        int tmp2 = n;
        n = m;
        m = tmp2;
    }

    static void four(){
        int[][] tmp = new int[m][n];

        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                tmp[m-1-j][i] = board[i][j];
            }
        }

        board = tmp;
        int tmp2 = n;
        n = m;
        m = tmp2;
    }

    static void five(){
        int[][] tmp = new int[n][m];
        int rh = n/2;
        int ch = m/2;

        for(int i=0;i<rh;i++){
            for(int j=0;j<ch;j++){
                tmp[i][j] = board[i+rh][j];
            }
        }

        for(int i=0;i<rh;i++){
            for(int j=ch;j<m;j++){
                tmp[i][j] = board[i][j-ch];
            }
        }

        for(int i=rh;i<n;i++){
            for(int j=ch;j<m;j++){
                tmp[i][j] = board[i-rh][j];
            }
        }

        for(int i=rh;i<n;i++){
            for(int j=0;j<ch;j++){
                tmp[i][j] = board[i][j + ch];
            }
        }

        board = tmp;
    }

    static void six(){
        int[][] tmp = new int[n][m];
        int rh = n/2;
        int ch = m/2;

        for(int i=0;i<rh;i++){
            for(int j=0;j<ch;j++){
                tmp[i][j] = board[i][j+ch];
            }
        }

        for(int i=0;i<rh;i++){
            for(int j=ch;j<m;j++){
                tmp[i][j] = board[i+rh][j];
            }
        }

        for(int i=rh;i<n;i++){
            for(int j=ch;j<m;j++){
                tmp[i][j] = board[i][j - ch];
            }
        }

        for(int i=rh;i<n;i++){
            for(int j=0;j<ch;j++){
                tmp[i][j] = board[i-rh][j];
            }
        }

        board = tmp;
    }


    static void print(){
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                System.out.print(board[i][j]+" ");
            }
            System.out.println();
        }
        System.out.println();
    }
}