[BaekJoon] 삼성 sw 역량 테스트(주사위 굴리기)

• 이 문제는 C++로 풀이를 작성하였습니다.

문제

 

예시

 

풀이

#include<iostream>
#include<vector>
using namespace std;


int n,m,r,c,num;
int arr[30][30];
int dir[5][2]={{0,0},{0,1},{0,-1},{-1,0},{1,0}};  // 방향 배열 (0,0)은 무시
int dice[3][3]={0}; // 주사위 
int back=0; // 주사위 바
    
struct node{
    int y,x;
    node(int y,int x):y(y),x(x){}
};

bool inside(int a,int b){
    return 0<=a && a<n && 0<=b && b<m;
}

int main(){
    cin>>n>>m>>r>>c>>num;
    vector<int> v;
    
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>arr[i][j];
        }
    }
    
    node nd=node(r,c);
    
    for(int i=0;i<num;i++){
        int d;
        cin>>d;
        int dy=dir[d][0]+nd.y;
        int dx=dir[d][1]+nd.x;
        
        if(inside(dy,dx)){
            nd=node(dy,dx);
            
            if(d==1){
                int temp=dice[1][2];
                dice[1][2]=dice[1][1];
                dice[1][1]=dice[1][0];
                dice[1][0]=back;
                back=temp;
            }
            else if(d==2){
                int temp=dice[1][0];
                dice[1][0]=dice[1][1];
                dice[1][1]=dice[1][2];
                dice[1][2]=back;
                back=temp;
            }
            else if(d==3){
                int temp=dice[0][1];
                dice[0][1]=dice[1][1];
                dice[1][1]=dice[2][1];
                dice[2][1]=back;
                back=temp;
            }
            else{
                int temp=dice[2][1];
                dice[2][1]=dice[1][1];
                dice[1][1]=dice[0][1];
                dice[0][1]=back;
                back=temp;
            }
            
            if(arr[dy][dx]==0){
                arr[dy][dx]=back;
            }
            else{
                back=arr[dy][dx];
                arr[dy][dx]=0;
            }
            v.push_back(dice[1][1]);
        }
        
        
    }
    for(int i=0;i<v.size();i++){
        cout<<v[i]<<' ';
    }
    
    
}​

• 추가로 궁금한 점이나 수정할 부분 있으면 댓글로 남겨주세요.