[BOJ 14891] 톱니바퀴

 각 톱니바퀴가 맞닿은 부분의 극이 반대라면 회전하고 같으면 회전하지 않는다. 이 때 입력으로 들어오는 모든 회전을 한 뒤 최종 톱니바퀴의 상태에 따라서 점수가 부여되는데 그 점수의 합을 구하는 문제이다.

 

문제 풀이 : 단순한 구현문제로 입력으로 들어온 회전을 할 때 마다 1,2,3,4 톱니바퀴의 회전방향을 저장한뒤 마지막에 다 돌려주었다.

 

 

#include <iostream>
#include <vector>
#include <string>
#include <cstring>
using namespace std;
 
int dir[5];
void rotate(vector <vector <char> > &b, int num, int rot) {
    if (rot == 1) {
        char tem = b[num].back();
        b[num].pop_back();
        b[num].insert(b[num].begin(), tem);
    }
    if (rot == -1) {
        char tem = b[num].front();
        b[num].erase(b[num].begin()+0);
        b[num].push_back(tem);
    }
}
void solve(vector <vector <char> > &b, int num, int rot) {
    
    for (int i = num; i-1 > 0; i--) {
        if (b[i][6] != b[i - 1][2]) {
            dir[i - 1] = dir[i]*-1;
            //rotate(b, num, rot);
        }
        else {
            break;
        }
    }
 
    for (int i = num; i + 1 <= 4; i++) {
        if (b[i][2] != b[i + 1][6]) {
            dir[i + 1] = dir[i]*-1;
        }
        else {
            break;
        }
    }
 
    for (int i = 1; i <= 4; i++) {
        if (dir[i] == 0) continue;
 
        if (dir[i] == 1) {
            rotate(b, i, 1);
        }
        if (dir[i] == -1) {
            rotate(b, i, -1);
        }
    }
 
}
int main() {
 
    vector <vector <char> > b(5);
    for (int i = 1; i <= 4; i++) {
        string s;
        getline(cin, s);
 
        for (int j = 0; j < s.length(); j++) {
            b[i].push_back(s[j]);
        }
    }
    int K; cin >> K;
 
    for (int i = 0; i < K; i++) {
        int num; int rot;
        cin >> num >> rot;
        memset(dir, 0, sizeof(dir));
        dir[num] = rot;
        solve(b,num,rot);
    }
 
    int ans = 0;
    if (b[1][0] == '1') ans++;
    if (b[2][0] == '1') ans += 2;
    if (b[3][0] == '1') ans += 4;;
    if (b[4][0] == '1') ans += 8;
 
    cout << ans << '\n';
 
    return 0;
}