삼성 SW 역량 테스트 기출 문제 : 톱니바퀴

https://www.acmicpc.net/problem/14891

14891번: 톱니바퀴

 

// 코드

import java.util.*;
import java.io.*; 
 
 
public class Main {   
  
  static int[] wheels = new int[4];
  
  // main
  public static void main(String[] args) throws Exception{
    
    // Please Enter Your Code Here
    BufferedWriter bw  = new BufferedWriter(new OutputStreamWriter(System.out));
    BufferedReader br  = new BufferedReader(new InputStreamReader(System.in)); 
    StringTokenizer st = null; 
    
    // 2진수로 읽어들인다.
    for(int i=0; i<4; ++i) wheels[i] = Integer.parseInt(br.readLine(), 2);
    
    // 주어진 조건에 맞추어 바퀴를 돌려준다.
    int K = Integer.parseInt(br.readLine());
    for(int i=0; i<K; ++i) {
      
      st = new StringTokenizer(br.readLine());
      int idx = Integer.parseInt(st.nextToken()) - 1;
      int dir = Integer.parseInt(st.nextToken());
      
      oper(idx, dir);
    }
    
    int ret = getScore();
    bw.write(String.valueOf(ret));
    
    bw.flush();
    bw.close();
    br.close();
  }
  
  
  // 바퀴의 회전해야할 모든 방향을 계산해준다.
  static void oper(int idx, int dir) {
    
    // 바퀴가 회전해야할 방향을 모두 가지고 있다. 
    // 1 : 시계 방향, 0 : 회전 안함, -1 : 반시계 방향
    int[] dirArr = new int[4];
    dirArr[idx] = dir;
    
    // 톱니바퀴 기준 왼쪽 톱니바퀴들의 방향
    for(int i=idx; i>0; --i) {
      if(isSame(i-1, i)) break; // 현재 톱니바퀴와 이전 톱니바퀴와 맞닿는 부분이 일치한다면 탈출
      dirArr[i-1] = -dirArr[i];
    }
    
    // 톱니바퀴 기준 오른쪽 톱니바퀴들의 방향
    for(int i=idx; i<3; ++i) {
      if(isSame(i, i+1)) break;
      dirArr[i+1] = -dirArr[i];
    }
    
    // 회전을 시켜준다.
    for(int i=0; i<4; ++i) rotate(i, dirArr[i]);
  }
  
  
  // 톱니바퀴 맞닿는 부분 체크
  static boolean isSame(int i, int j) {
    
    int orgBit = (wheels[i] & (1 << 5)) > 0 ? 1 : 0;
    int trgBit = (wheels[j] & (1 << 1)) > 0 ? 1 : 0;
    
    return orgBit == trgBit;
  }
  
  
  // 회전을 시켜준다.
  static void rotate(int idx, int dir) {
    
    // 시계방향 (->)
    if(dir == 1) {
      
      // 최하위 비트가 1인 경우, 최상위 비트를 1로 채워준다.
      if( (wheels[idx] & 1) > 0 ) {
        wheels[idx] >>>= 1;
        wheels[idx] |= (1 << 7);
      } else {
        wheels[idx] >>>= 1; 
      }
      
    } else if(dir == -1) { // 반시계 방향 (<-)
      
      // 최상위 비트가 1인 경우, 최하위 비트를 1로 채우준다.
      if( (wheels[idx] & (1 << 7)) > 0 ) {
        wheels[idx] <<= 1;
        wheels[idx] |= 1;
      } else {
        wheels[idx] <<= 1;
      }
        
    } 
  }
  
  
  // 답을 반환
  static int getScore() {
    int score = 0;
    
    score += (wheels[0] & (1 << 7)) > 0 ? 1 : 0;
    score += (wheels[1] & (1 << 7)) > 0 ? 2 : 0;
    score += (wheels[2] & (1 << 7)) > 0 ? 4 : 0;
    score += (wheels[3] & (1 << 7)) > 0 ? 8 : 0;
    
    return score;
  }
  
}