[Python]Queue - [백준 5430] AC

💡 문제 설명

🏅 난이도 골드5

R: 배열에 있는 수의 순서를 뒤집는 명령
D: 첫 번째 수를 버리는 명령 (빈 배열에서 사용하면 에러)

🛠 처음 풀이 방법

1. 테스트 케이스, 수행할 함수들, 수의 개수와 숫자 리스트를 deque 로 입력받는다.
2. 수행할 함수들을 하나씩 탐색하며 R이면 뒤집기를, D면 popleft()를 시행한다.

처음 생각한 풀이 방법은 매우 간단했다.
하지만 여러가지 주의해야할 내용이 몇 가지가 존재한다.

📌 주의사항

• deque로 입력 받을 시 큐가 비어있더라도 길이는 1로 인식되기 때문에 수의 개수가 0개라면 빈 리스트로 초기화를 해줘야 한다.
• 수행할 함수들은 최대 100,000개로 R함수가 나올때마다 매번 시행하면 시간초과가 발생한다.

✨ 해결 방법

• 위의 두 가지 사항을 주의해야 하므로 수의 개수 n이 0이라면 []로 초기화를 해줘야한다.
• R은 큐를 뒤집는 함수로 R이 짝수일 경우에는 수행하지 않아도 된다!
• R의 수를 세기 위해 flag라는 변수를 사용한다.

from collections import deque

T = int(input())  # 테스트 케이스 수

for _ in range(T):
    func = input()  # 수행할  함수들
    n = int(input())   # 배열에 들어있는 수의 개수 n
    num = deque(input()[1:-1].split(',')) # 배열에 들어있는 정수 리스트

    # deque는 비어 있어도 길이를 1로 인식해서 초기화 필요
    if n == 0:
        num = []

    # 시간초과 발생 방지 - R이 홀수일 때만 뒤집기
    flag = 0
    # 에러 여부 판단
    error = 0

    for i in func:
        if i == 'R':
            # 뒤집기 위한 flag 수
            flag += 1
        else:   
            # 첫 번 째 수 버리기
            if len(num) == 0:
                error = 1
                print('error')
                break
            else:
                # flag가 짝수면 뒤집을 필요가 없으므로 popleft
                if flag%2 == 0:
                    num.popleft()
                else:
                    # 뒤집어서 pop하는 것으로 그냥 pop
                    num.pop()

    if error == 0:
        # flag가 짝수면 원래 순서로 출력
        if flag%2 == 0:
            print('[' + ','.join(num) + ']')
        else:
            # 홀수면 뒤집힌 상태로 출력 
            num.reverse()
            print('[' + ','.join(num) + ']')

위에처럼 하면 시간초과도 나지 않고 잘 돌아가는 걸 확인할 수 있다!

다만 하나 궁금한 점이 있다 (이건 혹시 아는 분들은 댓글을 달아주세요🥲)

input()을 사용하는 것보다 sys.stdin.readline()을 사용하는 것이 시간을 더 단축시킬 수 있어서 이걸 사용했을 때는 계속 틀렸다고 떴었다,,,
설마 그거 때문이겠어 하고 코드를 여러가지로 수정해봤는데도 계속 틀렸다고 해서 결국에는 input()을 사용했더니 금새 해결됬다..!! 아직도 뭐가 문제인지 모르겠다... 혹시 아시는 분이 있다면 알려주세요🙇‍♀️

-아무튼 이 문제 마지막으로 queue 스텝 마무리👩🏻‍💻-