[Java Algorithm] 톱니바퀴 BOJ #14891

문제

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

14891번: 톱니바퀴

 

내 문제 풀이

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

public class Main {
	
	static char[][] ns = new char[4][]; // 각 기어들의 n, s극 정보
	static int[] dirs; // 각 기어들이 갖게 될 회전 방향 정보
	
	// 각 톱니의 회전 방향에 맞게 회전시켜주는 메소드
	public static void spin() {
		int tmp = 0;
		
		for(int i = 0; i < 4; i++) {
			// 시계 방향
			if(dirs[i] == 1) {
				tmp = ns[i][7];
				for(int j = 7; j > 0; j--) ns[i][j] = ns[i][j - 1];
				ns[i][0] = (char)tmp;
			}
			// 반시계 방향
			if(dirs[i] == -1) {
				tmp = ns[i][0];
				for(int j = 0; j < 7; j++) ns[i][j] = ns[i][j + 1];
				ns[i][7] = (char)tmp; 
			}
		}
	}

	public static void main(String[] args) throws Exception {
		BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;

		for(int i = 0; i < 4; i++) ns[i] = in.readLine().toCharArray();
		int num = Integer.parseInt(in.readLine());
		
		for(int i = 0; i < num; i++) {
			st = new StringTokenizer(in.readLine());
			int tobni = Integer.parseInt(st.nextToken()) - 1; // 회전을 시작하는 톱니의 인덱스
			dirs = new int[4];
			dirs[tobni] = Integer.parseInt(st.nextToken());
			
			// 왼쪽 톱니바퀴 검사
			for(int j = (tobni - 1); j >= 0; j--) {
				if(ns[j][2] == ns[j+1][6]) break;
				else dirs[j] = -dirs[j+1];
			}
			// 오른쪽 톱니바퀴 검사
			for(int j = (tobni + 1); j < 4; j++) {
				if(ns[j][6] == ns[j-1][2]) break;
				else dirs[j] = -dirs[j-1];
			}
			
			spin(); // 각 톱니 회전
		}
		
		// 결과 출력
		int answer = 0;
		int[] score = {1, 2, 4, 8};
		for(int i = 0; i < 4; i++) {
			if(ns[i][0] == '1') answer += score[i];
		}
		System.out.println(answer);
	}

}

👉 구현, 시뮬레이션 문제

 

[변수 설명]

• static char[][] ns : 입력되는 기어들의 n, s극 정보를 가진다. (n극 0, s극 1)
• static int[] dirs : 각 기어들이 회전해야 할 방향에 대한 정보를 가진다. (시계 1, 반시계 -1, 회전안함 0)
• tobni : 회전을 시작하는 톱니의 인덱스

 

[풀이 설명]

회전을 시작하는 톱니바퀴 기준, 왼쪽과 오른쪽을 검사하며 각 톱니바퀴마다 회전 유무 및 방향을 찾고,

각 톱니의 회전방향에 맞게 tmp 변수를 이용하여 배열의 요소들을 이동시켜주었다.

• 첫 번째로 회전하는 톱니바퀴를 -1하여 입력받았다. (인덱스로 활용하기 위해)
  그리고 dirs 배열의 tobni 자리에 해당 톱니바퀴의 회전 방향을 저장해주었다. (나머지 자리는 0)
• 회전 시작 톱니바퀴를 기준으로 왼쪽, 오른쪽을 차례대로 검사하였는데,
  항상 왼쪽 톱니바퀴의 2번 인덱스와 오른쪽 톱니바퀴의 6번 인덱스가 맞닿으므로
  이러한 점을 이용하여 맞닿은 부분의 값이 다를 경우 방향을 - 하여 dirs에 저장했다.
• 검사 완료 시 spin 메소드를 호출하여 각 톱니바퀴의 회전을 진행했다.
  dirs에 저장된 값이 0일 경우 회전하지 않는 톱니바퀴이므로 따로 조건을 지정하지 않았고,
  1(시계 방향)일 경우와 -1(반시계 방향)일 경우를 나눠서 회전을 시켜주었다.
• 시계 방향과 반시계 방향의 경우 각각 마지막, 첫 번째 인덱스를 tmp에 저장하고
  나머지 요소들을 한 칸씩 이동시키는 식으로 구현했다.

 

💡 피드백

• 풀이에 시간을 엄청 많이 소요했다. 크게 어려운 문제는 아니었지만,
  자바 언어가 익숙하지 않았던 부분도 있겠고,
  내가 꼼꼼히 문제를 풀이하지 못했던 문제도 있는 것 같다.
• 시뮬레이션 문제의 경우 생각해주어야 할 예외나 조건들을 명확히 지켜주는 것이 중요하므로
  더욱 꼼꼼히 메모하고 계획하여 문제를 풀이하는 습관을 길러야 하겠다.