[JAVA] 백준 14891번

www.acmicpc.net/problem/14891

14891번: 톱니바퀴

 

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
 
public class 톱니바퀴_bj_14891 {
    
    static int[][] wheels = new int[4][8];    // 톱니바퀴 상태
    static int[] turnDir = new int[4];        // 톱니바퀴를 회전시킬 방향 -1:역방향, 0:회전X, 1:시계방향
    
    static int K, sum;
 
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        
        // 톱니바퀴 상태
        for(int i=0; i<4; i++) {
            String temp = br.readLine();
            
            for(int j=0; j<8; j++) {
                wheels[i][j] = temp.charAt(j)-48;
            }
        }
        
        // 톱니바퀴 회전 횟수
        K = Integer.parseInt(br.readLine());
        
        // 회전 방법
        for(int i=0; i<K; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int wheelNum = Integer.parseInt(st.nextToken())-1;    // 톱니바퀴 index 때문에 -1해줌
            int dir = Integer.parseInt(st.nextToken());
            
            // 회전시킬 톱니바퀴 index에 회전 방향을 설정
            turnDir[wheelNum] = dir;
            
            // 회전시킬 톱니바퀴 왼쪽 톱니바퀴 탐색
            int temp = wheelNum;
            while(temp-1>=0) {
                if(wheels[temp][6] != wheels[temp-1][2]) {
                    turnDir[temp-1] = turnDir[temp]*-1;
                } else turnDir[temp-1] = 0;
                temp--;
            }
            
            // 회전시킬 톱니바퀴의 오른쪽 톱니바퀴 탐색
            temp = wheelNum;
            while(temp+1<4) {
                if(wheels[temp][2] != wheels[temp+1][6]) {
                    turnDir[temp+1] = turnDir[temp]*-1;
                } else turnDir[temp+1] = 0;
                temp++;
            }
            
            // 톱니바퀴 회전
            for(int j=0; j<4; j++) {
                if(turnDir[j] == -1) {
                    reverse(j);
                } else if(turnDir[j] == 1) {
                    turn(j);
                }
            }
            
        }
        
        // 점수계산 N극이면 0점, S극이면 2^i점
        for(int i=0; i<4; i++) {
            if(wheels[i][0] == 1)    sum+=Math.pow(2, i);
        }
        
        // 결과 출력
        System.out.println(sum);
    }
    
    // 시계 방향 회전
    static void turn(int num) {
        int temp = wheels[num][7];
        
        for(int i=6; i>=0; i--) {
            wheels[num][i+1] = wheels[num][i];
        }
        wheels[num][0] = temp; 
    }
    
    // 시계 반대 방향 회전
    static void reverse(int num) {
        int temp = wheels[num][0];
        
        for(int i=1; i<8; i++) {
            wheels[num][i-1] = wheels[num][i];
        }
        wheels[num][7] = temp;
    }
 
}