[Baekjoon] 5397_키로거

Silver III

문제(출처: https://www.acmicpc.net/problem/5397)

---

<키로거>

문제 

 

창영이는 강산이의 비밀번호를 훔치기 위해서 강산이가 사용하는 컴퓨터에 키로거를 설치했다. 며칠을 기다린 끝에 창영이는 강산이가 비밀번호 창에 입력하는 글자를 얻어냈다.
키로거는 사용자가 키보드를 누른 명령을 모두 기록한다. 따라서, 강산이가 비밀번호를 입력할 때, 화살표나 백스페이스를 입력해도 정확한 비밀번호를 알아낼 수 있다. 
강산이가 비밀번호 창에서 입력한 키가 주어졌을 때, 강산이의 비밀번호를 알아내는 프로그램을 작성하시오. 강산이는 키보드로 입력한 키는 알파벳 대문자, 소문자, 숫자, 백스페이스, 화살표이다.

 

 

입력

 

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한 줄로 이루어져 있고, 강산이가 입력한 순서대로 길이가 L인 문자열이 주어진다. (1 ≤ L ≤ 1,000,000) 강산이가 백스페이스를 입력했다면, '-'가 주어진다. 이때 커서의 바로 앞에 글자가 존재한다면, 그 글자를 지운다. 화살표의 입력은 '<'와 '>'로 주어진다. 이때는 커서의 위치를 움직일 수 있다면, 왼쪽 또는 오른쪽으로 1만큼 움직인다. 나머지 문자는 비밀번호의 일부이다. 물론, 나중에 백스페이스를 통해서 지울 수는 있다. 만약 커서의 위치가 줄의 마지막이 아니라면, 커서 및 커서 오른쪽에 있는 모든 문자는 오른쪽으로 한 칸 이동한다.

 

출력 

 

각 테스트 케이스에 대해서, 강산이의 비밀번호를 출력한다. 비밀번호의 길이는 항상 0보다 크다.

 

 

문제 풀이

   - 시간 초과 코드

import sys

if __name__=='__main__':
    t=int(sys.stdin.readline().strip())

    for i in range(t):
        x=sys.stdin.readline().strip()

        arr=[]
        idx=0
        for j in x:
            if j=='<':
                if idx!=0:
                    idx-=1
            elif j=='>':
                if idx<len(arr):
                    idx+=1
            elif j=='-':
                if idx>0:
                    idx-=1
                    arr.pop(idx)
                    if idx>0:
                        idx-=1
            else:
                left=arr[:idx]
                right=arr[idx:]
                left.append(j)
                arr=left+right
                idx+=1
                
        print(''.join(arr))

 

• >와 <일 때 index 이동
• - 일 때 pop
• 문자 입력일 때 list slice를 활용하여 append 후 붙이기

나름 index를 활용하여 시간을 줄일 수 있지 않을까 해서 사용한 건데 아니었나 보다 ㅎㅎ 문자 입력할 때 list를 계속해서 slice 하고 붙이는 것이 시간을 더 많이 사용하는 것 같다는 결론이 났다. (혼자만의 생각)

 

계속해서 발생하는 시간 초과로 결국 검색의 힘을 빌렸다. index가 아닌 커서 위치를 기준으로 왼쪽 오른쪽으로 나누어 list를 사용하는 것을 힌트로 얻을 수 있었다. 

 

 

- 통과한 코드

import sys

if __name__=='__main__':
    t=int(sys.stdin.readline().strip())

    for i in range(t):
        x=sys.stdin.readline().strip()

        left=[]
        right=[]
        for j in x:
            if j=='<': # 커서 왼쪽으로 이동
                if len(left)!=0:
                    right.append(left.pop())
            elif j=='>': # 커서 오르쪽으로 이동
                if len(right)>0:
                    left.append(right.pop())
            elif j=='-': # 백스페이스
                if len(left)!=0:
                    left.pop()
            else: # 글자 입력
                left.append(j)

        arr=left+list(reversed(right)) # right를 뒤집어서 left와 붙이기

        print(''.join(arr))

 

커서를 > 오른쪽으로 움직이면 왼쪽의 list 값을 하나 pop 하여 오른쪽 list에 넣는다.

< 왼쪽으로 움직이면 오른쪽의 list 값 하나를 pop 하여 왼쪽 list에 넣는다. 

- 이면 커서 왼쪽에 있는 값 하나를 삭제하므로 왼쪽 리스트 값을 하나 pop 한다. 

문자인 경우 커서 위치에 삽입하는 것이므로 왼쪽 리스트에 값을 추가한다.

 

마지막 왼쪽, 오른쪽 리스트를 합칠 때에도 실수하여 틀렸습니다가 한번 더 발생했었다.

리스트를 붙일 때 주의할 점은 오른쪽 list는 뒤집어서 붙여야 한다는 것이다.

 

• >와 <일 때 list pop 하여 반대쪽 list에 append
• - 일 때 왼쪽 list pop
• 글자 입력일 때 왼쪽 list append
• left+(right 뒤집기)

---

생각🤔

 

이 문제도 몇 달 전에 풀어보고 시간 초과를 해결 못해서 잠시 접어뒀던 문제이다.

아.. 다들 저렇게 푸는 것을 왜 난 못 푸는가?!

그래도 조금의 힌트를 보고 아차 하고는 문제를 풀 수 있었다.

파이팅 -!

 

---