21.08.17 - [BOJ] 15662. 톱니바퀴 (2)

[suhyunsim]

문제

• 15662

핵심 아이디어

• 톱니바퀴 문제와 비슷한데 4를 N으로 바꾸기

어려운 점, 실수

풀이

package main.java.com.poogle.BOJ.Q15662;

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        char[][] a = new char[n][8];
        for (int i = 0; i < n; i++) {
            a[i] = sc.next().toCharArray();
        }
        int k = sc.nextInt();
        while (k-- > 0) {
            int no = sc.nextInt() - 1;
            int dir = sc.nextInt();
            //각 톱니가 어떤 방향으로 회전하는지
            int[] d = new int[n];
            d[no] = dir;
            //왼쪽
            for (int i = no - 1; i >= 0; i--) {
                if (a[i][2] != a[i + 1][6]) {
                    d[i] = -d[i + 1];
                } else {
                    break;
                }
            }
            //오른쪽
            for (int i = no + 1; i < n; i++) {
                if (a[i - 1][2] != a[i][6]) {
                    d[i] = -d[i - 1];
                } else {
                    break;
                }
            }
            for (int i = 0; i < n; i++) {
                if (d[i] == 0) continue;
                if (d[i] == 1) {
                    //시계방향
                    char temp = a[i][7];
                    for (int j = 7; j >= 1; j--) {
                        a[i][j] = a[i][j - 1];
                    }
                    a[i][0] = temp;
                } else if (d[i] == -1) {
                    //반시계방향
                    char temp = a[i][0];
                    for (int j = 0; j < 7; j++) {
                        a[i][j] = a[i][j + 1];
                    }
                    a[i][7] = temp;
                }
            }
        }
        int ans = 0;
        for (int i = 0; i < n; i++) {
            if (a[i][0] == '1') {
                ans += 1;
            }
        }
        System.out.println(ans);
    }
}