[백준-5430] AC / Python

5430번: AC

import sys


def AC(command, data, n):
    command.replace("RR", "")
    l, r, d = 0, 0, True
    for c in command:
        if c == "R":
            d = not d
        elif c == "D":
            if d:
                l += 1
            else:
                r += 1
    if l + r <= n:
        answer = data[l : n - r]
        if d:
            return "[" + ",".join(answer) + "]"
        else:
            return "[" + ",".join(answer[::-1]) + "]"
    else:
        return "error"


input = sys.stdin.readline
for _ in range(int(input())):
    command = input()
    n = int(input())
    data = input().rstrip()[1:-1].split(",")
    if n == 0:
        data = []
    print(AC(command, data, n))

 

해설

명령어마다 pop을 수행하는 것이 아니라 왼쪽에서 지우는 수와 오른쪽에서 지우는 수를 센 후 일괄적으로 지웠다.

여기서 l과 r은 각 방향에서 지워지는 개수이고, d는 왼쪽인지 오른쪽인지 방향을 나타낸다.