[백준 5430] AC (Python)

2022. 8. 15. 18:48

문제 링크 #

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

문제 해설 #

Idea #

Implementation, Deque
문제에서 주어진대로 매번 배열을 뒤집으면 O(N^2)의 시간 복잡도로 시간 초과가 발생
배열에 영향을 주지 않으면서 R 함수를 처리하기 위해 상태 변수를 정의하고,
D 함수가 호출될 경우 배열의 상태에 따라 첫 번째 수를 버릴지 마지막 수를 버릴지 결정
마지막에 배열의 상태를 업데이트하고 정해진 형태로 결과를 출력

Time Complexity #

O(N) = 100,000

Data Size #

T: 1 <= int <= 100
p: 1 <= int <= 100,000
n: 1 <= int <= 100,000
arr: int(100) * n (like [x_1,…,x_n])

해설 코드 #

python

from collections import deque

for _ in range(int(input())):
    p = input()
    n = int(input())
    arr = deque(eval(input()))
    forward = True

    try:
        for cmd in p:
            if cmd == 'R':
                forward = not forward
            elif cmd == 'D':
                if forward:
                    arr.popleft()
                else:
                    arr.pop()
        arr = map(str,arr) if forward else map(str,reversed(arr))
        print(f'[{",".join(map(str,arr))}]')
    except IndexError:
        print('error')