[BOJ] 5397번 - 키로거 (Swift, C++)

문제

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

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

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

 

<문제 링크>

5397번: 키로거

---

입력

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

---

출력

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

---

문제 풀이

이번 문제는 문자 에디터와 같은 기능을 하는 코드를 작성하는 문제입니다. 

풀이로는 Swift 코드에서는 두 개의 배열을 사용하였고, C++ 코드에서는 연결 리스트를 활용하여 문제를 해결했습니다.

 

<swift>

배열을 이용할 경우 주의할 점은 '<' 처리 부분에서는 leftArr의 마지막 요소를 rightArr로 옮겨주고, '>' 처리 부분에서는 반대로 rightArr의 마지막 요소를 leftArr로 옮겨주면 됩니다. 이와 같은 기능은 스택을 이용할 수도 있고, 단순히 배열을 생성하여 메서드를 이용할 수도 있습니다.

 

또한, 계속해서 배열의 마지막 데이터만을 활용하기 때문에 범위를 벗어나는 접근은 하지 않습니다. 빈 배열일 때를 처리하는 방법으로는 isEmpty로 확인하는 방법과 옵셔널 타입을 리턴해주는 popLast 메서드를 활용할 수 있습니다.

 

<cpp>

연결 리스트를 사용할 때 주의할 점은 iterator가 리스트의 범위를 벗어나는 상황을 만들지 않는 것입니다.

리스트에 데이터를 추가하기 전에 L.begin()을 저장한 뒤, 커서가 범위를 벗어나는지 계속 확인해줍니다.

 

(빈 리스트에서 L.begin()과 L.end()는 같은 값을 가집니다. 또한, insert를 하면 해당 iter 앞쪽으로 데이터가 저장되기 때문에 처음에 저장한 begin 값은 계속해서 리스트의 마지막 인덱스를 가리킵니다.)

풀이 순서

1. Swift의 경우 두 개의 배열을 만들어주고, C++의 경우 라이브러리에서 지원해주는 List를 정의해줍니다.

2. 문제에서 요구하는 값을 조건에 맞게 저장해줍니다.

3. 조건에 맞게 저장된 값을 출력해줍니다.

---

풀이 코드

< Swift>

// Swift
// 풀이 1: 두 개의 배열 이용 (문자열로도 가능)
func solve(_ str: String) {
    
    var leftArr = [Character]()
    var rightArr = [Character]()
    
    for char in str {
        if char == "<" {
//            if let last = leftArr.popLast() {
//                rightArr.append(last)
//            }
            // 이 코드가 미세하게 빠름
            if !leftArr.isEmpty {
                rightArr.append(leftArr.removeLast())
            }
        } else if char == ">" {
//            if let last = rightArr.popLast() {
//                leftArr.append(last)
//            }
            if !rightArr.isEmpty {
                leftArr.append(rightArr.removeLast())
            }
        } else if char == "-" {
            let _ = leftArr.popLast()
        } else { // 문자 입력
            leftArr.append(char)
        }
    }
    
    print(String(leftArr) + String(rightArr.reversed()))
}

---

< C++>

// C++
#include <bits/stdc++.h>
using namespace std;

// 풀이 2: 연결 리스트 활용
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    
    int n;
    cin >> n;
    
    while(n--){
        list<char> L = {};
        
        auto iter = L.begin();  // 커서
        
        string cmd;
        cin >> cmd;
        
        for(auto c : cmd){  // 빈 리스트일 경우, begin()과 end()는 같은 값
            if(c == '>'){
                if(iter != L.end()) iter++;
            }
            else if(c == '<'){
                if(iter != L.begin()) iter--;
            }
            else if(c == '-'){
                if(iter != L.begin()){
                    iter--;
                    iter = L.erase(iter);  // iter가 erase되면서 다음 값의 iter를 리턴
                }
            }
            else{
                L.insert(iter, c);  // inter 앞에 c를 추가
            }
        }
        
        for(auto c : L) cout << c;
        cout << '\n';
        
    }
    
}

 

---

마무리

이번 문제에서는 stack을 완전히 구현하지 않고 메서드만을 이용하여 개념적으로만 접근했습니다. 

 

또한, Swift의 removeLast()는 @discardableResult를 이용한다는 것을 알게 되었습니다. 

(popLast()의 리턴 타입이 옵셔널이라 빈 배열을 예외처리할 때 자주 사용하였는데, 해당 메서드는 @discardableResult를 채택하고 있지 않습니다.) 

 

C++ 코드를 작성하면서 iterator 개념에 대해서 찾아보게 되었고, 메모리 주소에 대해 조금은 더 친숙해진 것 같습니다. 이후에는 Swift의 iterator protocol과 sequence에 대해서도 더 파고들면 좋을 것 같습니다.