[삼성SW테스트] 백준 14891번 - 톱니바퀴 (정답률 50%)

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

14891번: 톱니바퀴

삼성 SW 테스트 기출문제다. (시뮬레이션)

주어진 조건을 꼼꼼히 보는게 가장 관건인 시뮬레이션 유형이다.

K번 루프 내에서 주어지는 톱니의 번호와 방향을 기준으로 양 옆의 톱니 가운데 돌릴 톱니와 방향을 '미리' 정해주고 톱니들을 돌리는 것이 핵심이다.

 

즉, 루프에서 입력한 톱니와 방향을 돌리고 나서 양 옆의 톱니들을 살피는 것이 아니라, '미리' 톱니들의 회전을 파악하고 그대로 톱니들을 돌려야 한다.

 

조건대로 시뮬레이션 코드를 작성하면 아래와 같다. (0ms)

 

#include <iostream>
#include <vector>
#include <string>
using namespace std;
 
vector<vector<char> > topnis;
vector<pair<int, int> > Left, Right;
 
void rotate(int index, int dir){
    if (dir == 1){
        topnis[index][0] = topnis[index][8];
        for (int i = 7; i >= 1; i--){
            topnis[index][i + 1] = topnis[index][i];
        }
        topnis[index][1] = topnis[index][0];
    }
    else if (dir == -1){
        topnis[index][0] = topnis[index][1];
        for (int i = 1; i <= 7; i++){
            topnis[index][i] = topnis[index][i + 1];
        }
        topnis[index][8] = topnis[index][0];
    }
}
 
int getanswer(){
    int sum = 0;
    for (int i = 1; i <= 4; i++){
        if (i == 1){
            if (topnis[i][1] == '1')
                sum += 1;
        }
        else if (i == 2){
            if (topnis[i][1] == '1')
                sum += 2;
        }
 
        else if (i == 3){
            if (topnis[i][1] == '1')
                sum += 4;
        }
 
        else if (i == 4){
            if (topnis[i][1] == '1')
                sum += 8;
        }
    }
    return sum;
}
 
int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
 
    topnis.assign(5, vector<char>(9, ' '));
    string str;
    for (int i = 1; i <= 4; i++){
        cin >> str;
        for (int j = 1; j <= 8; j++){
            topnis[i][j] = str[j - 1];
        }
    }
    int K;
    int which, dir;
    cin >> K;
    for (int i = 0; i < K; i++){
        cin >> which >> dir;
        int lr, ll, rl, rr;
        lr = topnis[which][7];
        rl = topnis[which][3];
        int l_dir = -dir;
        int r_dir = -dir;
        for (int j = which - 1; j >= 1; j--){
            // 왼쪽 관찰.
            ll = topnis[j][3];
 
            if (ll == lr) break;
            else{
                Left.push_back(make_pair(j, l_dir));
                l_dir = l_dir * (-1);
                if (j - 1 < 1) break;
                else{
                    lr = topnis[j][7];
                }
            }
        }
 
 
        for (int j = which + 1; j <= 4; j++){
            // 오른쪽 관찰.
            rr = topnis[j][7];
 
            if (rl == rr) break;
            else{
                Right.push_back(make_pair(j, r_dir));
                r_dir = r_dir * (-1);
                if (j + 1 > 4) break;
                else{
                    rl = topnis[j][3];
                }
            }
        }
        rotate(which, dir);
        for (int j = 0; j < Left.size(); j++){
            int cur_which = Left[j].first;
            int cur_dir = Left[j].second;
            rotate(cur_which, cur_dir);
        }
        for (int j = 0; j < Right.size(); j++){
            int cur_which = Right[j].first;
            int cur_dir = Right[j].second;
            rotate(cur_which, cur_dir);
        }
        Left.clear();
        Right.clear();
    }
    cout << getanswer() << '\n';
    return 0;
}