[백준(BOJ)] 5397 키로거 - 스위프트 (Swift)

[Silver II] 키로거 - 5397

문제 링크

성능 요약

메모리: 90776 KB, 시간: 468 ms

분류

자료 구조, 연결 리스트, 스택

제출 일자

2023년 12월 29일 22:52:23

문제 설명

창영이는 강산이의 비밀번호를 훔치기 위해서 강산이가 사용하는 컴퓨터에 키로거를 설치했다. 며칠을 기다린 끝에 창영이는 강산이가 비밀번호 창에 입력하는 글자를 얻어냈다.

키로거는 사용자가 키보드를 누른 명령을 모두 기록한다. 따라서, 강산이가 비밀번호를 입력할 때, 화살표나 백스페이스를 입력해도 정확한 비밀번호를 알아낼 수 있다.

강산이가 비밀번호 창에서 입력한 키가 주어졌을 때, 강산이의 비밀번호를 알아내는 프로그램을 작성하시오. 강산이는 키보드로 입력한 키는 알파벳 대문자, 소문자, 숫자, 백스페이스, 화살표이다.

입력

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

출력

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

 

풀이

위 문제를 풀기 위해서 중요한 것은 화살표의 입력이라고 생각합니다. 이를 어떻게 구현할 지가 가장 큰 관건인데,
저는 임시 배열을 하나 만들어서 만약 '<'가 나왔다면 마지막 값을 임시 배열에 넣어놓고, '>'가 나왔다면 임시 배열에서 값을 꺼내는 방식을 사용하였습니다. 배열은 큐라고 생각하였습니다.

그리고 '-'가 나온다면 마지막 값을 제거하면 되는 문제입니다.

 

여기서 주의할 점은 임시 배열에 값이 남아있으면 안되기 때문에, 임시 배열의 값들을 다시 추가해주는 작업이 필요합니다. 이를 작업할 때에는 append를 통해 붙일 것이기 때문에 임시 배열을 뒤집어서 값을 추가해주어야 합니다.

 

다음은 코드입니다.

let n = Int(readLine()!)!

for _ in 0..<n {
    let input = readLine()!
    var leftStack = [Character]()   // 커서 왼쪽에 있는 값들
    var rightStack = [Character]()  //  커서 오른쪽에 있는 값들

    for char in input {
        if char == "<" {
            if !leftStack.isEmpty {
                rightStack.append(leftStack.removeLast())
            }
        } else if char == ">" {
            if !rightStack.isEmpty {
                leftStack.append(rightStack.removeLast())
            }
        } else if char == "-" {
            if !leftStack.isEmpty {
                leftStack.removeLast()
            }
        } else {
            leftStack.append(char)
        }
    }

    leftStack.append(contentsOf: rightStack.reversed()) // 오른쪽 커서 스택은 뒤집혀있기 때문에 뒤집어서 추가해줘야 함
    print(String(leftStack))
}