[JAVA] 백준 14891번 톱니바퀴

문제

14891번: 톱니바퀴

입력

• 첫째 줄에 1번 톱니바퀴의 상태, 둘째 줄에 2번 톱니바퀴의 상태, 셋째 줄에 3번 톱니바퀴의 상태, 넷째 줄에 4번 톱니바퀴의 상태가 주어진다. 상태는 8개의 정수로 이루어져 있고, 12시방향부터 시계방향 순서대로 주어진다. N극은 0, S극은 1로 나타나있다.
• 다섯째 줄에는 회전 횟수 K(1 ≤ K ≤ 100)가 주어진다. 다음 K개 줄에는 회전시킨 방법이 순서대로 주어진다. 각 방법은 두 개의 정수로 이루어져 있고, 첫 번째 정수는 회전시킨 톱니바퀴의 번호, 두 번째 정수는 방향이다. 방향이 1인 경우는 시계 방향이고, -1인 경우는 반시계 방향이다.

출력

• 총 K번 회전시킨 이후에 네 톱니바퀴의 점수의 합을 출력한다. 점수란 다음과 같이 계산한다.
  • 1번 톱니바퀴의 12시방향이 N극이면 0점, S극이면 1점
  • 2번 톱니바퀴의 12시방향이 N극이면 0점, S극이면 2점
  • 3번 톱니바퀴의 12시방향이 N극이면 0점, S극이면 4점
  • 4번 톱니바퀴의 12시방향이 N극이면 0점, S극이면 8점

입력 예제

10101111
01111101
11001110
00000010
2
3 -1
1 1

출력 예제

7

문제 풀이

• 비트 연산 문제인가 했는데 구현 문제였다. 가장 먼저 규칙을 찾아야 한다.
  • 시계 10101111 -> 11010111 -> 맨 뒤 숫자 앞으로
    반시계 00000010 -> 00000001 -> 맨 앞 숫자 맨 뒤로
  • 첫번째 톱니는 2번 인덱스와 비교
    이전 톱니 2번 인덱스 <-> 6번 인덱스
    현재 톱니 2번 인덱스 <-> 다음 톱니 6번 인덱스
    마지막 톱니는 6번 인덱스 
  • 위와 같은 규칙을 조건에 맞게 잘 구현해주기만 하면 된다.
  • 회전의 정보를 입력 받을 때 마다 각 함수를 실행시켜주었고, 회전 정보가 들어올 때마다 회전정보를 저장하는 배열을 만들어 정보를 저장하고, 저장된 정보대로 시계 방향 또는 반시계 방향으로 회전시켜주었다 !

코드

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

public class boj14891 {
	
	static int[][] arr;
	static int result;

	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		arr = new int[4][8];
		
		for (int i = 0; i < 4; i++) {
			String s = br.readLine();
			for (int j = 0; j < 8; j++) {
				arr[i][j] = s.charAt(j) - '0';
			}
		}
//		print(arr);
		
		// 회전 횟수 입력
		int n = Integer.parseInt(br.readLine());
		for (int i = 0; i < n; i++) {
			StringTokenizer st = new StringTokenizer(br.readLine());
			
			// 회전할 톱니, 방향
			int gear = Integer.parseInt(st.nextToken()) - 1;
			int dir = Integer.parseInt(st.nextToken());
			
			// 회전 방향 정해서 d배열에 넣기
			int[] d = new int[4];
			d[gear] = dir;
			// 회전 방향 정하고
			chkRotate(gear, dir , d);
//			for (int j = 0; j < 4; j++) {
//				System.out.println(d[j]);
//			}
			// 회전하기
			rotate(d);
//			print(arr);
		}
//		print(arr);
		calcGear(arr);
		System.out.println(result);
	}

	// 점수 계산하기
	private static void calcGear(int[][] arr) {
		if (arr[0][0] == 1) result += 1;
		if (arr[1][0] == 1) result += 2;
		if (arr[2][0] == 1) result += 4;
		if (arr[3][0] == 1) result += 8;
		
	}

	// 저장된 회전방향 정보대로 회전하기 1:시계, -1:반시계
	private static void rotate(int[] d) {
		for (int i = 0; i < d.length; i++) {
			// 시계 방향 회전
			if (d[i] == 1) {
				int tmp = arr[i][7];
				for (int j = 7; j >= 1; j--) {
					arr[i][j] = arr[i][j - 1];
				}
				arr[i][0] = tmp;
			}
			// 반시계 방향 회전
			else if (d[i] == -1) {
				int tmp = arr[i][0];
				for (int j = 0; j <= 6; j++) {
					arr[i][j] = arr[i][j + 1];
				}
				arr[i][7] = tmp;
			}
		}
		
	}

	// 기어의 회전방향 정하는 함수
	private static void chkRotate(int gear, int dir, int[] d) {
		// 현재 기어번호
		int nowLeft = gear;
		// 이전 방향 체크
		for (int i = gear - 1; i >= 0; i--) {
			// 방향이 다를 때 회전
			if (arr[nowLeft][6] != arr[i][2]) {
				d[i] = -d[nowLeft];
				nowLeft--;
			}
			else break;
		}
		
		// 현재 기어번호
		int nowRight = gear;
		// 이후 방향 체크
		// 방향이 다를 때 회전
		for (int i = gear + 1; i < 4; i++) {
			if (arr[nowRight][2] != arr[i][6]) {
				d[i] = -d[nowRight];
				nowRight++;
			}
			else break;
		}
		
	}

	private static void print(int[][] arr) {
		for (int i = 0; i < 4; i++) {
			for (int j = 0; j < 8; j++) {
				System.out.print(arr[i][j]);
			}
			System.out.println();
		}
		
	}
}