[코드트리] 정육면체 한번 더 굴리기 - 시뮬레이션 (기출문제)

삼성 SW 역량테스트 2021 하반기 기출

정육면체 한번 더 굴리기

1이상 6이하 중 임의의 숫자가 그려진 n * n 격자판에 한 면이 1 * 1 크기인 정육면체를 놓여져 있습니다. 해당 격자판에서 정육면체를 굴리려합니다.

다음과 같이 4 * 4 크기의 격자판이 주어졌다고 가정해보겠습니다.

처음 정육면체의 각 면에는 1부터 6까지의 숫자가 다음과 같이 쓰여져 있고 m번에 걸쳐 주사위를 계속 1칸씩 굴리게 됩니다. 이때, 마주보는 면에 적혀있는 숫자의 합은 정확히 7입니다.

주사위는 항상 초기에 격자판의 1행 1열에 놓여져 있고, 처음에는 항상 오른쪽으로 움직입니다.

 

(중략)

 

n * n 크기의 격자판의 상태가 주어졌을 때, m번 진행하며 얻게되는 점수의 총 합을 구하는 프로그램을 작성해보세요.

입력 형식

첫째줄에 격자 크기 n, 그리고 굴리는 횟수 m이 공백을 사이에 두고 주어집니다.

둘째줄부터 n+1번째 줄까지 격자판에 적혀있는 수 n개가 한 줄에 하나씩 공백을 사이에 두고 주어집니다.

• 2 ≤ n ≤ 20
• 1 ≤ m ≤ 1000
• 1 ≤ 말판에 쓰여진 수 ≤ 6

출력 형식

주사위를 m번 굴렸을 때 까지 얻게되는 총 점수의 합을 출력합니다.

 

---

 

일단 알고리즘 자체는 굉장히..단순한 문제이다.

bfs 혹은 dfs로 탐색하고 주사위만 잘 굴려주면 되는 문제.

그런데 주사위 돌릴때 숫자를 어떻게 저장하고 갱신해야할지 감이 안잡혀서 해설을 참고했다.

 

중간에 실수 하나 때문에 몇십분 날려먹었는데,,

정신차리자!!!!

 

 

나의 풀이

#include <iostream>
#include <queue>

using namespace std;

int n;
int m;
int map[21][21];
pair<int,int> dice;
int moveDirection = 0; // 0=오른쪽,1=아래,2=왼쪽,3=위
int Bottom = 6; // 주사위 바닥숫자 초기값
int Up = 1;
int Right = 3;
int Front = 2;
int dx[4] = {1,0,-1,0};
int dy[4] = {0,1,0,-1};
int Point = 0;

void checkDirection(int num){
    // cout << "Bottom = "<<Bottom<<"\n";
    if(Bottom > num) {
        if(moveDirection<3) moveDirection++;
        else moveDirection = 0;
    } else if(Bottom < num){
        if(moveDirection>0) moveDirection--;
        else moveDirection = 3;
    } else return;
}

void checkBFS(int x, int y){
    // 현재 좌표 받아야함
    int visited[21][21]={0};
    queue<pair<int,int>> q;
    q.push({x,y});
    visited[x][y] = 1;
    int tmp = map[x][y];
    int cnt = 1;
    while(!q.empty()){
        pair<int,int> a = q.front();
        q.pop();
        for(int i=0; i<4; ++i){
            int nx = a.first + dx[i];
            int ny = a.second + dy[i];
            if(nx<0 || ny<0 || nx>n || ny>n) continue;
            if(map[nx][ny] != tmp) continue;
            if(visited[nx][ny]) continue;
            q.push({nx,ny});
            visited[nx][ny] = 1;
            cnt++;
        }
    }
    
    Point += tmp * cnt;
    // cout << tmp << " * " << cnt << " => "<< Point<<"\n";
    checkDirection(tmp);
}

void moveDice(){
    // cout << "\n"<<dice.first <<" "<< dice.second << "\n";
    int tmp = Up;
    if(moveDirection==0){ // 오른쪽
        if(dice.second + 1 > n){
            moveDirection = 2;
            moveDice();
            return;
        }
        else {
            // cout << "right\n";
            Up = 7 - Right;
            Right = tmp;
            Bottom = 7 - Up;
            dice.second++;
        }
    }
    else if(moveDirection==1){ // 아래
        if(dice.first + 1 > n){
            moveDirection = 3;
            moveDice();
            return;
        }else {
            // cout << "down\n";
            Up = 7-Front;
            Front = tmp;
            Bottom = 7-Up;
            dice.first ++;
        }
    }
    else if(moveDirection==2){ // 왼쪽
        if(dice.second - 1 < 1){
            moveDirection = 0;
            moveDice();
            return;
        }else {
            // cout << "left\n";
            Up = Right;
            Right = 7-tmp;
            Bottom = 7-Up;
            dice.second --;
        }
    }
    else if(moveDirection==3){ // 위
        if(dice.first - 1 < 1){
            moveDirection = 1;
            moveDice();
            return;
        }else {
            // cout << "up\n";
            Up = Front;
            Front = 7-tmp;
            Bottom = 7-Up;
            dice.first --;
        }
    }
    
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);

    cin>>n>>m;
    for(int i=1; i<=n; ++i){
        for(int j=1; j<=n; ++j){
            cin >> map[i][j];
        }
    }
    dice = {1,1};
    for(int i=0; i<m; ++i){
        moveDice();
        checkBFS(dice.first, dice.second);
    }
    cout << Point;
    return 0;
}