백준 5430 AC

문제

👀 문제 사이트 : https://www.acmicpc.net/problem/5430

풀이

난이도 : Gold V

이 문제는 배열에서 popleft와 reverse를 사용해야하는 문제이다. 하지만, 문제에서 reverse가 주어질 때마다 배열의 reverse를 수행하면 시간 초과가 발생할 것이다. 따라서 reverse가 입력되는 것은 변수로 저장만 해두기로 하고, pop을 할 때 reverse의 값에 따라 popleft() 또는 pop()을 진행하였다. 마지막으로 reverse가 입력된 횟수가 홀수이면 마지막에 reverse를 해주고 결과 출력을 진행하였다.

✔ 주의할 점

• 이 문제에 경우에는 입력방식과 출력방식이 특이하므로 주의하여야 한다.
• python에서는 입력을 할 때, rstrip을 사용하지 않을 경우 '틀렸습니다'가 나오게 되므로 문자열 입력과 파싱에서 주의하여야 한다.
• python에서 이 문제와 같이 출력이 복잡할 경우에 .join()을 사용하면 쉽게 구현이 가능하다.

코드

import sys
from collections import deque

t = int(input())

for _ in range(t):
    commands = sys.stdin.readline().rstrip()
    n = int(input())
    arr = sys.stdin.readline().rstrip()[1:-1].split(",")
    q = deque(arr)

    isTrue = True
    reverse = 0

    if n == 0:
        q = []
    
    for command in commands:
        if command == "R":
            reverse += 1
        elif command == "D":
            if len(q) == 0:
                isTrue = False
                break

            if reverse % 2 == 0:
                q.popleft()
            else:
                q.pop()

    if isTrue:
        if reverse % 2 == 0:
            print("[" + ",".join(q) + "]")
        else:
            q.reverse()
            print("[" + ",".join(q) + "]")
    else:
        print("error")