[Code Tree] 정육면체 굴리기 (삼성 SW 역량테스트 2016 하반기 1번)

---

문제

정육면체 굴리기

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

---

풀이

• 해당 풀이에서 가장 중요한 부분은 정육면체에 대한 기준을 정하는 것.
• 정육면체에는 총 6개의 면이 존재하는데 해당 면들에 넘버링을 해주는 부분이 개인적으로 생각을 요구했음...
• 진행하다보니 그냥 4가지 이동하는 경우에 대해서 면들의 위치만을 바꿔주는 rotateCube(int dir) 메서드를 구현해 해결할 수 있었음

---

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

class Pair {
    int x;
    int y;

    public Pair(int x, int y) {
        this.x = x;
        this.y = y;
    }
}

public class Main {
    static int N, M, K;
    static Pair cubePos;
    static int[] cube;
    static int[][] dxy = {{0,1},{0,-1},{-1,0},{1,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());
        cubePos = new Pair(Integer.parseInt(st.nextToken()),Integer.parseInt(st.nextToken()));
        K = Integer.parseInt(st.nextToken());
        board = new int[N][M];

        cube = new int[6];

        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 k = 0; k < K; k++) {
            int dir = Integer.parseInt(st.nextToken()) - 1;

            int nx = cubePos.x + dxy[dir][0];
            int ny = cubePos.y + dxy[dir][1];

            if(nx < 0 || ny < 0 || nx >= N || ny >= M) continue;

            rotateCube(dir);

            if(board[nx][ny] == 0) {
                board[nx][ny] = cube[5];
            }
            else if(board[nx][ny] != 0) {
                cube[5] = board[nx][ny];
                board[nx][ny] = 0;
            }

            cubePos.x = nx;
            cubePos.y = ny;

            System.out.println(cube[0]);
        }
    }


    static void rotateCube(int dir) {
        if(dir == 0) {
            cube = new int[]{cube[4], cube[1], cube[0], cube[3], cube[5], cube[2]};
        }
        else if(dir == 1) {
            cube = new int[]{cube[2], cube[1], cube[5], cube[3], cube[0], cube[4]};
        }
        else if(dir == 2) {
            cube = new int[]{cube[1], cube[5], cube[2], cube[0], cube[4], cube[3]};
        }
        else if(dir == 3) {
            cube = new int[]{cube[3], cube[0], cube[2], cube[5], cube[4], cube[1]};
        }
    }
}

• rotateCube 메서드에서는 문제에서 제시된대로 동(0), 서(1), 북(2), 남(3) 의 방향으로 큐브를 돌릴 때 현재의 위치를 기준으로 그 다음 면들의 순서는 어떻게 되는지에 대한 과정을 포함함.
• 아래는 문제의 내용을 바탕으로 각 면에대해 인덱스를 넘버링한 모습을 그림(?)으로 표현해봤는데.. 마우스로 그림 어케그리나요

• 아무튼 일단 인덱스를 부여하고 해당 인덱스별로 동,서,북,남 왔다리갔다리 굴렸을 때 번호를 바꿔주는 함수가 rotateCube라는 말
• 그리고 main 메서드에서는 입력으로 주어지는 이동 순서에 맞춰 이동하고, 이동할때마다 칸에서 수행해야되는 숫자복사작업을 진행.