[Python Algorithm] AC BOJ #5430

문제

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

5430번: AC

 

내 문제 풀이

import sys

t = int(sys.stdin.readline().strip())
for _ in range(t):
    p = sys.stdin.readline().strip()
    n = int(sys.stdin.readline().strip())
    x = sys.stdin.readline().strip()
    lst = list(x[1:len(x)-1].split(","))

    rev = False
    front = 0
    back = 0
    error = False

    for i in p:
        if i == "R":
            if rev:
                rev = False
            else:
                rev = True
        else:
            if rev:
                back += 1
            else:
                front += 1
            if front + back > n:
                error = True
        if error:
            break

    if error:
        print("error")
    else:
        if rev:
            print("[" + ",".join(lst[::-1][back:(n-front)]) + "]")
        else:
            print("[" + ",".join(lst[front:(n-back)]) + "]")

👉 문자열 문제

R이나 D 명령을 진짜로 하나씩 연산할 경우 시간이 너무 오래걸릴 것이므로

R의 유무로 현재 문자열이 뒤집혀있는지 아닌지를 판단하고, (rev 변수)

D 명령이 있을 시 앞에서 삭제할 것인지 뒤에서 삭제할 것인지 체크해주었다. (front, back 변수)

그리고 마지막에, 앞에서 몇 개, 뒤에서 몇 개의 문자를 삭제할 것인지,

문자열을 뒤집어서 출력해야 할 것인지를 모두 반영하여서 결과로 출력해주었다.

 

💡 피드백

결과를 출력할 때, 진짜 리스트를 그대로 출력하는 것이아니라

"[", "]", "," 문자를 이용하여 리스트 모양으로 출력해야 한다는 점만 주의하면 될 것 같다.