글 읽기 - [파이썬] 시간초과 도움 부탁드립니다

batsy_22 1년 전 0

스택 2개로 pop, append 사용해서 구현했는데 계속 5%에서 시간초과 뜨네요 ㅠㅠ

어떻게하면 복잡도를 줄일 수 있을까요..

import sys

sentence = sys.stdin.readline().strip()
m = int(sys.stdin.readline().strip())
list_command = [sys.stdin.readline().strip() for _ in range(m)]

position = len(sentence)  # 초기 커서 위치
stack_char = [elem for str in sentence.split() for elem in str]  # 문자열 스택

for command in list_command:
    tempStack = []

    if command == 'L':
        if stack_char and position != 0:
            position -= 1  # 커서를 왼쪽으로 한 칸 이동
    elif command == 'D':
        if stack_char and position < len(stack_char):
            position += 1  # 커서를 오른쪽으로 한 칸 이동
    elif command == 'B':  # 커서 왼쪽에 있는 문자 삭제
        if stack_char:
            if position == len(stack_char):
                stack_char.pop()
                position -= 1
            elif position == 0:
                continue
            else:  # 문장 중간에 커서가 위치
                idx = position  # tempStack까지 삽입할 커서 위치
                while idx == position:
                    tempStack.append(stack_char.pop())
                    idx -= 1

                if stack_char:  # 문자열 스택이 비어있지 않을 경우
                    stack_char.pop()
                    position -= 1
                while tempStack:
                    stack_char.append(tempStack.pop())

    else:  # P $ 문자를 커서 왼쪽에 삽입
        if position == len(stack_char):  # 문장의 맨 뒤에 커서가 위치
            stack_char.append(command[2])
        else:  # 문장 중간에 커서가 위치
            idx = len(stack_char) - 1  # tempStack까지 삽입할 인덱스 체크

            while idx > position:
                tempStack.append(stack_char.pop())
                idx -= 1

            tempStack.append(stack_char.pop())
            stack_char.append(command[2])  # 명령어로 삽입할 문자

            while tempStack:
                stack_char.append(tempStack.pop())

        position += 1

print(''.join(stack_char))

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

[파이썬] 시간초과 도움 부탁드립니다

batsy_22 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말