[BOJ/백준] 14891 톱니바퀴 - 파이썬

문제

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

14891번: 톱니바퀴

 

문제풀이

n번 바퀴의 2번 index는 n+1번 바퀴의 6번 index와 인접하고

n번 바퀴의 6번 index는 n-1번 바퀴의 2번 index와 인접한다.

 

이 두 값을 먼저 update 한 후에,

a번을 기준으로 왼쪽/오른쪽으로 2개의 for문을 통해 나머지 바퀴들을 탐색한다.

a번과 바로 인접한 바퀴의 경우, a번과 반대 방향으로 돌아가야 하기 때문에

b에 -1을 곱해준 값을, 그 다음 바퀴는 또 다시 -1을 곱한 b 값을 연산해 주어야 한다.

deque에 포함된 rotate 함수를 통해 index 변환을 더 쉽게 해주었다.

 

• rotate 연산

   rotate() : 가장 오른쪽 값을 pop 한 뒤에 appendleft로 추가해준다.
               괄호 안에 인자가 들어가면 그 값만큼 이동
               -값이 들어가면 가장 왼쪽 값을 popleft 한 후에 append 해준다.

 

 

코드

from collections import deque
​
saw=[deque(map(int,input())) for _ in range(4)]
k=int(input())
how=[]
​
for i in range(k):
    how.append(list(map(int,input().split())))
    how[i][0]-=1
​
for a,b in how:
    tmp1=saw[a][2]
    tmp2=saw[a][6]
    k=b
    for i in range(a-1,0-1,-1):
        if saw[i][2]!=tmp2:
            tmp2=saw[i][6]
            saw[i].rotate(-1*k)
            k*=-1
        else:
            break
    k=b
    for i in range(a+1,4):
        if saw[i][6]!=tmp1:
            tmp1=saw[i][2]
            saw[i].rotate(-1*k)
            k*=-1
        else:
            break
    
    saw[a].rotate(b)
result=0
for i in range(4):
    if saw[i][0]==1:
        result+=(2**i)
print(result)

 

어려웠던 점 / 느낀점

처음에는 rotate 대신 함수를 직접 구현하여 실행했다. 예시는 다 통과하길래 제출했는데 제출하자마자 틀렸다.

구글링하다가 rotate 함수가 있다는 것을 발견!!