[BOJ] #14891 톱니바퀴

시간 제한	메모리 제한	정답 비율
2 초	512 MB	49.598%

 

14891번: 톱니바퀴

 

방법

톱니바퀴들을 gears 라는 배열에 저장해 두고 회전시킬 톱니바퀴를 gear 라고 했을 때, gear 기준으로 왼쪽 톱니바퀴들과 오른쪽 톱니바퀴들을 나눠서 처리한 다음 마지막에 gear 번째 톱니바퀴를 회전시키도록 하였다.

시뮬레이션 문제답게 회전할 때마다 배열을 수정할 수 있지만 가장 먼저 떠올랐던 '인덱스로 접근'하는 방법으로 구현하였다. 4개의 바퀴의 12시 방향을 저장해두는 배열 idx[] 를 두어, 회전할 때마다 idx[톱니바퀴 번호]를 갱신시켰다.

• 시계방향이라면 idx[gear] = (idx[gear]+7) % 8;
• 반 시계방향으로 회전이라면 idx[gear] = (idx[gear]+1) % 8; 

나머지 연산자를 쓰는 이유는 조건문 없이 바로 갱신이 가능하기 때문이다.

메모리	시간
1116 KB	0 ms

#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
int gears[5][10], idx[5], t, gear, dir, preidx, predir, ans; 
char c[10];
int main(){
    memset(gears, -1, sizeof(gears));
    for (int i = 0; i < 4; i++) {
        scanf("%s", c);
        for (int j = 0; j < 8; j++) {
            gears[i][j] = c[j] - '0';
        }
    }
    for (int i = 0; i < 4; i++) idx[i] = 0; // 12시 방향 인덱스는 0으로 초기화.
 
    scanf("%d", &t);
    while (t--) {
        scanf("%d %d", &gear, &dir);
        // 같지 않으면 돌아감.
        if (gear > 1 && gears[gear - 1][(idx[gear - 1] + 6) % 8] != gears[gear - 2][(idx[gear - 2] + 2) % 8]) {
            predir = dir;
            for (int i = gear - 2; i >= 0; i--) {
                preidx = idx[i];
                if (predir == 1) { // 이전이 시계방향이었다면 반시계방향으로 돌아야함.
                    idx[i] = (idx[i] + 1) % 8;
                    predir = -1;
                }
                else { // 이전이 반시계방향이었다면 시계방향으로 돌아야함.
                    idx[i] = (idx[i] + 7) % 8;
                    predir = 1;
                }
                // preidx 와 왼쪽 톱니바퀴를 비교.
                if (gears[i][(preidx + 6) % 8] == gears[i - 1][(idx[i - 1] + 2) % 8]) break;
            }
        }
        if (gear < 4 && gears[gear - 1][(idx[gear - 1] + 2) % 8] != gears[gear][(idx[gear] + 6) % 8]) {
            predir = dir;
            for (int i = gear; i <= 3; i++) {
                preidx = idx[i];
                if (predir == 1) { // 이전이 시계방향이었다면 반시계방향으로 돌아야함.
                    idx[i] = (idx[i] + 1) % 8;
                    predir = -1;
                }
                else { // 이전이 반시계방향이었다면 시계방향으로 돌아야함.
                    idx[i] = (idx[i] + 7) % 8;
                    predir = 1;
                }
                // preidx 와 오른쪽 톱니바퀴를 비교.
                if (gears[i][(preidx + 2) % 8] == gears[i + 1][(idx[i + 1] + 6) % 8]) break; // 양쪽 끝 부분도 돌아가게 하기 위하여 -1로 초기화 시켜두고 같은지로 판별.
            }
        }
        // gear -1 번째 톱니 idx 변경.
        if (dir == 1) { // 시계방향
            idx[gear - 1] = (idx[gear - 1] + 7) % 8;
        }
        else { // 반시계방향
            idx[gear - 1] = (idx[gear - 1] + 1) % 8;
        }
    }
    int sq = 1;
    for (int i = 0; i < 4; i++) { // S극 은 1로 나타나있고 
        ans += gears[i][idx[i]] * sq;    sq *= 2;
    }
    printf("%d", ans);
 
    return 0;
}

 

 

입력이 띄어쓰기 없이 주어지는 부분을 문자열로 받아 int로 변환하는 방법을 택했는데, 형식지정자로 처리할 수 있는 방법이 있다. 그리고 gears 배열을 1번째 인덱스에서부터 넣어 좀 더 직관적이도록 바꿔보았다.

#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
int gears[5][10], idx[5], t, gear, dir, preidx, predir, ans; 
int main(){
    memset(gears, -1, sizeof(gears));
    for (int i = 1; i <= 4; i++) {
        for (int j = 0; j < 8; j++) {
            scanf("%1d", &gears[i][j]);
        }
    }
    scanf("%d", &t);
    while (t--) {
        scanf("%d %d", &gear, &dir);
        // 왼쪽 톱니바퀴
        if (gears[gear][(idx[gear] + 6) % 8] != gears[gear - 1][(idx[gear - 1] + 2) % 8]) {
            predir = dir;
            for (int i = gear -1; i > 0; i--) {
                preidx = idx[i];
                if (predir == 1) { // 이전이 시계방향이었다면 반시계방향으로 돌아야함.
                    idx[i] = (idx[i] + 1) % 8; predir = -1;
                }
                else { // 이전이 반시계방향이었다면 시계방향으로 돌아야함.
                    idx[i] = (idx[i] + 7) % 8;    predir = 1;
                }
                // preidx 와 왼쪽 톱니바퀴를 비교.
                if (gears[i][(preidx + 6) % 8] == gears[i - 1][(idx[i - 1] + 2) % 8]) break;
            }
        }
        // 오른쪽 톱니바퀴
        if (gears[gear][(idx[gear] + 2) % 8] != gears[gear+1][(idx[gear+1] + 6) % 8]) {
            predir = dir;
            for (int i = gear + 1; i <= 4; i++) {
                preidx = idx[i];
                if (predir == 1) { // 이전이 시계방향이었다면 반시계방향으로 돌아야함.
                    idx[i] = (idx[i] + 1) % 8;
                    predir = -1;
                }
                else { // 이전이 반시계방향이었다면 시계방향으로 돌아야함.
                    idx[i] = (idx[i] + 7) % 8;
                    predir = 1;
                }
                // preidx 와 오른쪽 톱니바퀴를 비교.
                if (gears[i][(preidx + 2) % 8] == gears[i + 1][(idx[i + 1] + 6) % 8]) break; // 양쪽 끝 부분도 돌아가게 하기 위하여 -1로 초기화 시켜두고 같은지로 판별.
            }
        }
        // gear 번째 톱니 12시 방향 idx 변경.
        if (dir == 1) { // 시계방향
            idx[gear] = (idx[gear] + 7) % 8;
        }
        else { // 반시계방향
            idx[gear] = (idx[gear] + 1) % 8;
        }
    }
    int sq = 1;
    for (int i = 1; i <= 4; i++) { // S극 은 1 
        ans += gears[i][idx[i]] * sq;    sq *= 2;
    }
    printf("%d", ans);
 
    return 0;
}