백준 14891 : 톱니바퀴

문제 링크입니다 https://www.acmicpc.net/problem/14891

14891번: 톱니바퀴

총 8개의 톱니를 가지고 있는 톱니바퀴 4개가 아래 그림과 같이 일렬로 놓여져 있습니다.

 

 

톱니바퀴를 총 K번 회전시키는데, 톱니바퀴가 회전할 때, 시계방향 혹은 반 시계방향으로 회전시킬 수 있는데 서로 맞닿은 극이 다르면 옆에 있는 톱니바퀴도 반대 방향으로 같이 회전하게 됩니다. 톱니바퀴의 초기 상태와 톱니바퀴를 회전시킨 방법이 주어졌을 때, 최종 톱니바퀴의 상태를 구하는 문제였습니다.

 

 

[소스코드]

 

#include<iostream>
using namespace std;
 
int k, ans, gear[5][10];    // 12시부터 시계방향, N극은 0, S극은 1
 
// 회전시킬 톱니바퀴의 번호(배열의 인덱스)와 방향을 입력받고
// 그 방향으로 회전한 결과를 gear 배열에 update 해주는 함수
void getResult(int num, int dir) {
 
    if (dir == 1) {
        // 시계 방향으로 회전합니다
        int temp = gear[num][7];
        for (int i = 7; i > 0; i--) {
            gear[num][i] = gear[num][i - 1];
        }
        gear[num][0] = temp;
    }
    else {
        // 반 시계 방향으로 회전합니다
        int temp = gear[num][0];
        for (int i = 0; i < 7; i++) {
            gear[num][i] = gear[num][i + 1];
        }
        gear[num][7] = temp;
    }
}
 
int main(void) {
    for (int i = 0; i < 4; i++)
        for (int j = 0; j < 8; j++)
            scanf("%1d", &gear[i][j]);
 
    cin >> k;
    while (k--) {
        int a, b;
        scanf("%d%d", &a, &b);
 
        // 0 ~ 3번까지 톱니바퀴 4개를 어느 방향으로 회전할지
        // 결정하여 담을 result 배열
        int result[4] = { 0, };
        int num = a - 1;
 
        // 입력받은 톱니바퀴는 b 방향으로 회전
        result[num] = b;
 
        // 현재 톱니바퀴의 오른쪽 톱니바퀴의
        // 회전 방향을 결정합니다
        for (int i = num; i < 3; i++)
            if (gear[i][2] != gear[i + 1][6])
                result[i + 1] = -result[i];
 
        // 현재 톱니바퀴의 왼쪽 톱니바퀴의 
        // 회전 방향을 결정합니다
        for (int i = num; i > 0; i--)
            if (gear[i][6] != gear[i - 1][2])
                result[i - 1] = -result[i];
 
        // 결정된 각 방향으로 톱니바퀴를 회전합니다
        for (int i = 0; i < 4; i++)
            if (result[i] != 0)
                getResult(i, result[i]);
    }
    for (int i = 0; i < 4; i++) {
        if (gear[i][0] == 0) continue;
        else {
            if (i == 0) ans += 1;
            else if (i == 1) ans += 2;
            else if (i == 2) ans += 4;
            else if (i == 3) ans += 8;
        }
    }
    cout << ans << '\n';
    return 0;
}