[Java] BOJ 14891번 톱니바퀴

14891번: 톱니바퀴

---

풀이

시계방향과 반시계 방향으로 회전시키는 함수를 각각 만들어 해결했습니다.

이때, 톱니바퀴를 반시계 방향으로 회전시키면 붙어있는 톱니바퀴들은 시계방향으로 회전하고 반대로 시계방향으로 회전시키면 붙어있는 톱니바퀴들은 반시계 방향으로 회전하는 성질을 이용했습니다.

또한, visited 배열을 사용하여 이미 회전시킨 톱니를 다시 방문하지 않도록 처리했습니다.

---

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
	public static int K;
	public static boolean[] visited;
	public static int[][] gears = new int[4][8];
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = null;
		
		for (int i = 0; i < 4; ++i) {
			String input = br.readLine();
			
			for (int j = 0; j < 8; ++j) {
				gears[i][j] = input.charAt(j) - '0';
			}
		}
		
		K = Integer.parseInt(br.readLine());
		
		for (int i = 0; i < K; ++i) {
			st = new StringTokenizer(br.readLine(), " ");
			
			int gearNumber = Integer.parseInt(st.nextToken()) - 1;
			int direction = Integer.parseInt(st.nextToken());
			
			visited = new boolean[4];
			
			if (direction == 1) {
				turn(gearNumber);
			}
			else if (direction == -1) {
				reverseTurn(gearNumber);
			}
		}
		
		int answer = 0;
		for (int i = 0; i < 4; ++i) {
			if (gears[i][0] == 1) {
				answer += Math.pow(2, i);
			}
		}
		
		System.out.println(answer);
	}
	
	public static void turn(int n) {
		visited[n] = true;
		
		// 왼쪽에 있는 톱니 회전
		// 톱니의 맞닿은 극이 다를때만 회전해야함
		if (n - 1 >= 0 && !visited[n - 1] && gears[n - 1][2] != gears[n][6]) {
			reverseTurn(n - 1);
		}
		
		// 오른쪽에 있는 톱니 회전
		// 톱니의 맞닿은 극이 다를때만 회전해야함
		if (n + 1 < 4 && !visited[n + 1] && gears[n + 1][6] != gears[n][2]) {
			reverseTurn(n + 1);
		}
		
		int temp = gears[n][7];
		
		for (int i = 7; i > 0; i--) {
			gears[n][i] = gears[n][i - 1];
		}
		
		gears[n][0] = temp;		
	}
	
	public static void reverseTurn(int n) {
		visited[n] = true;
		
		// 왼쪽에 있는 톱니 회전
		// 톱니의 맞닿은 극이 다를때만 회전해야함
		if (n - 1 >= 0 && !visited[n - 1] && gears[n - 1][2] != gears[n][6]) {
			turn(n - 1);
		}
		
		// 오른쪽에 있는 톱니 회전
		// 톱니의 맞닿은 극이 다를때만 회전해야함
		if (n + 1 < 4 && !visited[n + 1] && gears[n + 1][6] != gears[n][2]) {
			turn(n + 1);
		}
		
		int temp = gears[n][0];
		
		for (int i = 0; i < 7; i++) {
			gears[n][i] = gears[n][i + 1];
		}
		
		gears[n][7] = temp;
	}
}

---

pyo7410/Algorithm