[백준 5430 / Python] AC

🧑🏻‍💻 문제링크

문제풀이

문제 자체는 크게 어렵지 않았던 문제이다. 다만, 시간복잡도를 생각해야 한다는 점이 문제의 핵심인 것 같다. 이전에는 어떻게든 문제만 풀면 되는 줄 알았는데 효율성도 고려해서 풀어야 겠다는 생각이 들었다.

문제를 풀면서 배웠던 것은 sys를 활용해야 겠다는 점. 그리고 flag와 같이 어떻게 보면 문제를 더 쉽게 접근할 수 있다는 점이다. 처음부터는 어렵겠지만 연습하는 버릇을 들여야겠다.

첫 번재 코드는 시간효율성에서 초과가 났던 코드이고 두 번째 코드는 다른 사람의 코드를 참고하여 풀었던 코드이다. 시간날 때 마다 비교해봐야 겠다.

코드1

def func(p, arr_temp):
    temp = arr_temp

    if temp == '':
        return temp

    # 정수형 배열로 만듬
    arr_case_to_list = list(map(int, arr_temp.split(',')))
    
    for i in p:
        if i == 'R':
            arr_case_to_list = arr_case_to_list[::-1]
        else:
            if len(arr_case_to_list) < 1:
                break
            else:
                arr_case_to_list.pop(0)
    return arr_case_to_list
            

T = int(input())    # 테스트 케이스
answer = []

for _ in range(T):
    
    answer = []
    # 수행할 함수 p
    p = input()
    # 배열에 들어있는 수의 개수
    n = int(input())
    # 배열 입력(문자열 형태)
    arr_case = input()

    # 정수형 배열로 만들기 위해 문자열의 '[', ']'부분을 제거
    arr_case_temp = arr_case[1:-1]
    
    answer = func(p, arr_case_temp)

    if len(answer) == 0:
        print("error")
    else:
        print(answer)

코드2

import sys
from collections import deque

T = int(sys.stdin.readline())

for _ in range(T):
   
   p = sys.stdin.readline()        # 수행할 함수
   n = int(sys.stdin.readline())   # 배열에 들어있는 수의 개수
   arr = sys.stdin.readline().rstrip()[1:-1].split(",")    # 배열

   queue = deque(arr)              # 덱 생성

   reverse_flag = 0                # R 플래그
   error_flag = 0                  # 에러 플래그

   # 배열에 아무것도 없을 때
   if n == 0:
       queue = []

   for cmd in p:
       # "R"일 때 뒤집기 플래그를 1추가
       if cmd == "R":
           reverse_flag += 1
       # "D"일 때 큐에 아무것도 없으면 에러
       elif cmd == "D":
           if len(queue) < 1:
               error_flag = 1
               print("error")
               break
           # 뒤집기가 짝수이면 원래대로 첫 번째에서 빼줌
           else:
               if reverse_flag % 2 == 0:
                   queue.popleft()
               # 홀수이면 뒤에서 빼는게 뒤집에서 빼는 것이랑 같은 원리
               else:
                   queue.pop()

   if error_flag == 0:
       if reverse_flag % 2 == 0:
           print("[" + ",".join(queue) + "]")
       else:
           queue.reverse()
           print("[" + ",".join(queue) + "]")

결과