[BaekJoon] 삼성 sw 역량 테스트(톱니바퀴)

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

문제

 

예시

 

풀이

#include <iostream>
#include <queue>
#include <utility>
#include <cmath>
#include <string>
using namespace std;

int main(){
    
    int sum=0;
    int arr[4][8];      // 톱니바퀴 배열
    
    for(int i=0;i<4;i++){
        string str;
        cin>>str;
        for(int j=0;j<8;j++){
            arr[i][j]=str[j]-'0';
        }
    }
    
    int k;
    cin>>k;
    
    for(int i=0;i<k;i++){
        
        int s,d;
        cin>>s>>d;
        
        int check[4]={0};
        queue<pair<int,int>> q;
        // 톱니바퀴 번호는 0부터 시작
        q.push(make_pair(s-1,d));
        check[s-1]=d;
        
        while(!q.empty()){
            int idx=q.front().first;
            int dir=q.front().second;
            q.pop();
            
            // 0이면 오른쪽만 확인
            if(idx==0){
                if(arr[idx][2]!=arr[idx+1][6] && check[idx+1]==0){      // 맞물린 톱니가 다르고, 방문 톱니가 아닐경우
                    q.push(make_pair(idx+1,-1*dir));
                    check[idx+1]=-1*dir;
                }
            }
            // 3이면 왼쪽만 확인
            else if(idx==3){
                if(arr[idx][6]!=arr[idx-1][2] && check[idx-1]==0){      // 맞물린 톱니가 다르고, 방문 톱니가 아닐경우
                    q.push(make_pair(idx-1,-1*dir));
                    check[idx-1]=-1*dir;
                }
            }
            // 그 외 양쪽 확인
            else{
                if(arr[idx][2]!=arr[idx+1][6] && check[idx+1]==0){      // 맞물린 톱니가 다르고, 방문 톱니가 아닐경우
                    q.push(make_pair(idx+1,-1*dir));
                    check[idx+1]=-1*dir;
                }
                if(arr[idx][6]!=arr[idx-1][2] && check[idx-1]==0){      // 맞물린 톱니가 다르고, 방문 톱니가 아닐경우
                    q.push(make_pair(idx-1,-1*dir));
                    check[idx-1]=-1*dir;
                }
            }
        }
        
        // 톱니 정렬
        for(int j=0;j<4;j++){
            if(check[j]==1){
                int temp=arr[j][7];
                for(int k=7;k>=1;k--){
                    arr[j][k]=arr[j][k-1];
                }
                arr[j][0]=temp;
            }
            else if(check[j]==-1){
                int temp=arr[j][0];
                for(int k=0;k<7;k++){
                    arr[j][k]=arr[j][k+1];
                }
                arr[j][7]=temp;
            }
        }
    }
    
    for(int i=0;i<4;i++){
        if(arr[i][0]==1){
            sum+=pow(2,i);
        }
    }
    cout<<sum;
}

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