190408 BOJ 14891번 톱니바퀴

시뮬레이션

- 1번, 4번은 순서대로 보면 되지만 2번, 3번은 양 옆을 따로 봐야한다.

1번 톱니바퀴 이동 명령 1->2->3->4

4번 톱니바퀴 이동 명령 4->3->2->1

2번 톱니바퀴 이동 명령 2->1 // 2->3->4

3번 톱니바퀴 이동 명령 3->4 // 3->2->1

- 1번과 3번은 같은 방향으로 움직인다.

- 2번과 4번은 같은 방향으로 움직인다.

소스 코드

#include <iostream>
#include <cmath>
#define FOR(i,a,b) for(int i = a; i < b; i++)
using namespace std;
int K;
int arr[5][8];
void move1(int x) { //시계방향이동
    int tmp = arr[x][7];
    for (int i = 7; i > 0; i--) {
        arr[x][i] = arr[x][i - 1];
    }
    arr[x][0] = tmp;
}
void move2(int x) { //반시계방향이동
    int tmp = arr[x][0];
    FOR(i,0,7) {
        arr[x][i] = arr[x][i + 1];
    }
    arr[x][7] = tmp;
}
int main() {
    FOR(i, 1, 5) FOR(j, 0, 8) scanf("%1d", &arr[i][j]);
    cin >> K;
    int num, d;
    while (K--) {
        cin >> num >> d; //톱니바퀴 번호, 방향
        switch (num)
        {
        case 1:
            if (arr[1][2] != arr[2][6]) { // 2번의 이동여부
                if (arr[2][2] != arr[3][6]) { //3번의 이동여부
                    if (arr[3][2] != arr[4][6])    d == 1 ? move2(4) : move1(4); //4번의 이동여부
                    d == 1 ? move1(3) : move2(3);
                }
                d == 1 ? move2(2) : move1(2);
            }
            d == 1 ? move1(1) : move2(1);
            break;
        case 2:
            if (arr[1][2] != arr[2][6]) d == 1 ? move2(1) : move1(1);
            if (arr[2][2] != arr[3][6]) {
                if (arr[3][2] != arr[4][6]) d == 1 ? move1(4) : move2(4);
                d == 1 ? move2(3) : move1(3);
            }
            d == 1 ? move1(2) : move2(2);
            break;
        case 3:
            if (arr[3][2] != arr[4][6]) d == 1 ? move2(4) : move1(4); //4번의 이동여부
            if (arr[2][2] != arr[3][6]) { //2번의 이동여부
                if (arr[1][2] != arr[2][6]) d == 1 ? move1(1) : move2(1);
                d == 1 ? move2(2) : move1(2);
            }
            d == 1 ? move1(3) : move2(3);
            break;
        case 4:
            if (arr[3][2] != arr[4][6]) { // 3번의 이동여부
                if (arr[2][2] != arr[3][6]) { //2번의 이동여부
                    if (arr[1][2] != arr[2][6]) d == 1 ? move2(1) : move1(1); //1번의 이동여부
                    d == 1 ? move1(2) : move2(2);
                }
                d == 1 ? move2(3) : move1(3);
            }
            d == 1 ? move1(4) : move2(4);
            break;
        }
    }
    int ans = 0;
    FOR(i, 1, 5) if (arr[i][0] == 1) ans += pow(2, i - 1);
    cout << ans;
}