[백준/C++] 14891 - 톱니바퀴

풀이 시간: 52m20s45

시간 복잡도: O(K)

공간 복잡도: O(1)

문제

입력

첫째 줄에 1번 톱니바퀴의 상태, 둘째 줄에 2번 톱니바퀴의 상태, 셋째 줄에 3번 톱니바퀴의 상태, 넷째 줄에 4번 톱니바퀴의 상태가 주어진다. 상태는 8개의 정수로 이루어져 있고, 12시방향부터 시계방향 순서대로 주어진다. N극은 0, S극은 1로 나타나있다. 

다섯째 줄에는 회전 횟수 K(1 ≤ K ≤ 100)가 주어진다. 다음 K개 줄에는 회전시킨 방법이 순서대로 주어진다. 각 방법은 두 개의 정수로 이루어져 있고, 첫 번째 정수는 회전시킨 톱니바퀴의 번호, 두 번째 정수는 방향이다. 방향이 1인 경우는 시계 방향이고, -1인 경우는 반시계 방향이다.

출력

총 K번 회전시킨 이후에 네 톱니바퀴의 점수의 합을 출력한다. 점수란 다음과 같이 계산한다. 

• 1번 톱니바퀴의 12시방향이 N극이면 0점, S극이면 1점 
• 2번 톱니바퀴의 12시방향이 N극이면 0점, S극이면 2점 
• 3번 톱니바퀴의 12시방향이 N극이면 0점, S극이면 
• 4점 4번 톱니바퀴의 12시방향이 N극이면 0점, S극이면 8점

문제 풀이

본 문제는 단순 시뮬레이션 문제이기 때문에 문제에서 주어진 조건대로 하나씩 맞춰가면서 풀어보았다.

문제에 조건이 자세히 나와있으므로 추가 설명은 주석으로 대체한다.

전체 코드

#include <iostream>
#include <deque>
using namespace std;
int num = 0, dir = 0;
deque<int> w1, w2, w3, w4; // 12시 방향부터 시계방향 순서대로 삽입

void forward(int w) { // 시계 회전
    switch (w) {
        case 0: // w1
            w1.push_front(w1.back());
            w1.pop_back();
            break;
        case 1: // w2
            w2.push_front(w2.back());
            w2.pop_back();
            break;
        case 2: // w3
            w3.push_front(w3.back());
            w3.pop_back();
            break;
        case 3: // w4
            w4.push_front(w4.back());
            w4.pop_back();
            break;
    }
}

void backward(int w) { // 반시계 회전
    switch (w) {
        case 0: // w1
            w1.push_back(w1.front());
            w1.pop_front();
            break;
        case 1: // w2
            w2.push_back(w2.front());
            w2.pop_front();
            break;
        case 2: // w3
            w3.push_back(w3.front());
            w3.pop_front();
            break;
        case 3: // w4
            w4.push_back(w4.front());
            w4.pop_front();
            break;
    }
}

void rotateWheel() {
    // 톱니는 전부 동시에 회전하기 때문에 각 톱니의 좌우 값을 따로 저장해줌
    // forward, backward 함수에 의해 값이 변동되기 때문
    int r1 = w1[2], l1 = w1[6], r2 = w2[2], l2 = w2[6];
    int r3 = w3[2], l3 = w3[6], r4 = w4[2], l4 = w4[6];
    if (dir == 1) { // 시계
        switch (num) {
            case 1: // w1
                forward(0); // w1 시계
                if (r1 != l2) {
                    backward(1); // w2 반시계
                    if (r2 != l3) {
                        forward(2); // w3 시계
                        if (r3 != l4)
                            backward(3); // w4 반시계
                    }
                }
                break;
            case 2: // w2
                forward(1); // w2 시계
                if (r1 != l2) backward(0); // w1 반시계
                if (r2 != l3) {
                    backward(2); // w3 반시계
                    if (r3 != l4) forward(3); // w4 시계
                }
                break;
            case 3: // w3
                forward(2); // w3 시계
                if (r3 != l4) backward(3); // w4 반시계
                if (r2 != l3) {
                    backward(1); // w2 반시계
                    if (r1 != l2) forward(0); // w1 시계
                }
                break;
            case 4: // w4
                forward(3); // w4 시계
                if (r3 != l4) {
                    backward(2); // w3 반시계
                    if (r2 != l3) {
                        forward(1); // w2 시계
                        if (r1 != l2)
                            backward(0); // w1 반시계
                    }
                }
                break;
        }
    }
    else { // 반시계
        switch (num) {
            case 1:
                backward(0); // w1 반시계
                if (r1 != l2) {
                    forward(1); // w2 시계
                    if (r2 != l3) {
                        backward(2); // w3 반시계
                        if (r3 != l4)
                            forward(3); // w4 시계
                    }
                }
                break;
            case 2:
                backward(1); // w2 반시계
                if (r1 != l2) forward(0); // w1 시계
                if (r2 != l3) {
                    forward(2); // w3 시계
                    if (r3 != l4) backward(3); // w4 반시계
                }
                break;
            case 3:
                backward(2); // w3 반시계
                if (r3 != l4) forward(3); // w4 시계
                if (r2 != l3) {
                    forward(1); // w2 시계
                    if (r1 != l2) backward(0); // w1 반시계
                }
                break;
            case 4:
                backward(3); // w4 반시계
                if (r3 != l4) {
                    forward(2); // w3 시계
                    if (r2 != l3) {
                        backward(1); // w2 반시계
                        if (r1 != l2)
                            forward(0); // w1 시계
                    }
                }
                break;
        }
    }
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    for (int i = 0; i < 4; ++i) { // 톱니바퀴 상태 삽입
        string str; cin >> str;
        for (int j = 0; j < str.size(); ++j) {
            int x = str[j] - '0';
            switch (i) {
                case 0: w1.push_back(x); break;
                case 1: w2.push_back(x); break;
                case 2: w3.push_back(x); break;
                case 3: w4.push_back(x); break;
            }
        }
    }
    int K; cin >> K;
    while (K--) { // 이동 횟수
        cin >> num >> dir;
        rotateWheel();
    }
    int score = 0; // S극일 때만 점수 추가
    if (w1.front() == 1) score += 1;
    if (w2.front() == 1) score += 2;
    if (w3.front() == 1) score += 4;
    if (w4.front() == 1) score += 8;
    cout << score;
    return 0;
}

 

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

14891번: 톱니바퀴