[ 백준 14891번 ] 톱니바퀴

1. 문제

14891번: 톱니바퀴

 

2. 풀이

단순 구현 문제이지만 문제에서 유의해야 할 점은 바로 톱니바퀴가 연쇄적으로 상태가 바뀌는 것이 아니라는 점이다. 이게 무슨 말이냐면 아래 이미지에서 1번 톱니바퀴를 회전시킨 결과에 따라서 2, 3, 4번 톱니바퀴의 회전이 결정되는 것이 아니라는 말이다. 즉, 아래 이미지와 같이 정적인 상태에서 회전 여부가 결정이 되고 최종적으로 한 번에 회전하게 된다는 것이다. 

 

회전에 대한 결과가 다음 회전에 대해 영향을 미치지 않게 구현하기 위해서는 단순 반복문이 아닌 재귀를 통해서 해결해야 한다. 세부 구현은 아래 코드를 참고하자.

 

3. 소스코드

public class Main {
    static final char N = '0', S = '1';

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        Factory factory = new Factory();
        for (int i = 0; i < 4; i++) {
            factory.setGear(i, br.readLine());
        }

        int K = Integer.parseInt(br.readLine());
        while (K-- > 0) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int idx = Integer.parseInt(st.nextToken()) - 1;
            int dir = Integer.parseInt(st.nextToken());
            factory.rotateGear(idx, idx, dir);
        }

        bw.write(factory.getScore() + "");
        bw.close();
        br.close();
    }

    static class Factory {
        Gear[] gears = new Gear[4];

        void setGear(int idx, String status) {
            this.gears[idx] = new Gear(status);
        }

        int getScore() {
            int tot = 0;
            for (int i = 0; i < 4; i++) {
                tot += gears[i].getGearValue(0) == '0' ? 0 : Math.pow(2, i);
            }
            return tot;
        }

        void rotateGear(int startIdx, int idx, int dir) {
            if (dir == 0) return;

            if (startIdx <= idx && idx < 3) {
                if (gears[idx].getGearValue(2) != gears[idx + 1].getGearValue(6)) {
                    rotateGear(startIdx, idx  + 1, -dir);
                } else {
                    rotateGear(startIdx, idx + 1, 0);
                }
            }

            if (0 < idx && idx <= startIdx) {
                if (gears[idx].getGearValue(6) != gears[idx - 1].getGearValue(2)) {
                    rotateGear(startIdx, idx - 1, -dir);
                } else {
                    rotateGear(startIdx, idx - 1, 0);
                }
            }

            gears[idx].rotate(dir);
        }
    }

    static class Gear {
        char[] status;

        public Gear(String line) {
            this.status = line.toCharArray();
        }

        char getGearValue(int idx) {
            return status[idx];
        }

        void rotate(int dir) {
            if (dir == 0) return;
            if (dir == 1) {
                char temp = status[7];
                for (int i = 7; i >= 1; i--) {
                    status[i] = status[i - 1];
                }
                status[0] = temp;
            } else {
                char temp = status[0];
                for (int i = 0; i <= 6; i++) {
                    status[i] = status[i + 1];
                }
                status[7] = temp;
            }
        }

    }
}

https://github.com/kwanik-kor/BOJ/blob/main/src/simulation/sml_14891_gear.java

GitHub - kwanik-kor/BOJ: Baekjoon Online Judge - explanation of solved problems and complete code