29-H0ngJu

[H0ngJu]

🔗 문제 링크

톱니바퀴

✔️ 소요된 시간

50M

✨ 수도 코드

문제 요약
톱니바퀴를 돌린 후, 조건에 맞게 점수를 출력하라. 맞닿아 있는 부분이 극이 서로 같으면 반대 방향으로 돌아가고, 같으면 돌아가지 않는다.

톱니 바퀴가 돌아가는 규칙은 다음과 같습니다.

1. 맞닿아 있는 극이 반대면, 서로 반대방향으로 회전한다.
2. 맞닿아 있는 극이 같으면, 회전하지 않는다.

그렇다면 현재 톱니바퀴를 기준으로 왼쪽, 오른쪽 바퀴들의 N,S극들을 조사하여, 어떤 방향으로 회전해야하는지 찾아주면 됩니다.

예를 들어서,

요런 경우에 3번 바퀴를 반대방향으로 돌리면,

2번 바퀴는 회전하지 않기 때문에, 1번 바퀴는 확인할 필요가 없습니다.
4번 바퀴는 3번 바퀴의 회전 반대 방향으로 돌려주면 됩니다.

이를 구현하기 위해서

현재 톱니 바퀴 기준으로, 오른쪽, 왼쪽을 나누어서

시계 반대방향으로 돌아가야 하는 톱니바퀴를 저장하는 left_check와
시계 방향으로 돌아가야 하는 톱니바퀴를 저장하는 right_check을 따로 두어서 체크해주면 됩니다.

for i in range(n, 0, -1): # 현재 톱니바퀴 기준 왼쪽
        if wheels[i][6] != wheels[i-1][2]:
            if left_check[i]:
                right_check[i-1] = True
            if right_check[i]:
                left_check[i-1] = True
        else:
            break
    
    for i in range(n,3): # 현재 톱니바퀴 기준 오른쪽
        if wheels[i][2] != wheels[i+1][6]:
            if left_check[i]:
                right_check[i+1] = True
            if right_check[i]:
                left_check[i+1] = True
        else:
            break

체크된 shift에 따라서, 톱니바퀴를 돌려주면 끝! 입니다

for i in range(4):
        if left_check[i]:
            leftshift(wheels[i])
        
        if right_check[i]:
            rightshift(wheels[i])

📚 새롭게 알게된 내용

[tgyuuAn]

import sys

def input(): return sys.stdin.readline().rstrip()
gears = [input() for _ in range(4)]

def rotate_gear(now_number, dir, visited):
    visited.add(now_number)
    
    origin = gears[now_number][:]
    
    if(dir == 1): gears[now_number] = gears[now_number][-1] + gears[now_number][0:-1]
    elif(dir == 0): return
    else: gears[now_number] = gears[now_number][1] + gears[now_number][2:] + gears[now_number][0]
    
    left_gear = now_number - 1
    if left_gear >= 0 and left_gear not in visited:
        left_dir = 0 if(origin[6] == gears[left_gear][2]) else dir * -1
        rotate_gear(left_gear, left_dir, visited)
    
    right_gear = now_number + 1
    if right_gear <= 3 and right_gear not in visited:
        right_dir = 0 if(origin[2] == gears[right_gear][6]) else dir * -1
        rotate_gear(right_gear, right_dir, visited)
 
    return

K = int(input())
for _ in range(K):
    gear_number, dir = map(int, input().split())
    rotate_gear(gear_number-1, dir, set())

answer = 0
for idx in range(4): 
    if gears[idx][0] == "1": answer += 2**(idx)
    
print(answer)

이 문제 재밌는데,

돌아가고 안돌아가는 조건이 한 턴 돌아간 뒤에 판별하는 건 줄 알았는데,

돌아가기 전에 걸로 판별하는 거였네요...... 여기서 시간 쓴 듯.... 아무리 봐도 이상해서..