방과후
ALGORITHM NOTE약 2분

BOJ 1406 - 에디터

커서는 바쁜데 손은 더 바빠

#algorithm#boj#silver#data-structures#stack
아카이브로 돌아가기

문제 링크

문제

한 줄로 된 간단한 에디터를 구현하려고 한다. 이 편집기는 영어 소문자만을 기록할 수 있는 편집기로, 최대 600,000글자까지 입력할 수 있다.

이 편집기에는 '커서'라는 것이 있는데, 커서는 문장의 맨 앞(첫 번째 문자의 왼쪽), 문장의 맨 뒤(마지막 문자의 오른쪽), 또는 문장 중간 임의의 곳(모든 연속된 두 문자 사이)에 위치할 수 있다. 즉 길이가 L인 문자열이 현재 편집기에 입력되어 있으면, 커서가 위치할 수 있는 곳은 L+1가지 경우가 있다.

이 편집기가 지원하는 명령어는 다음과 같다.

명령어설명
L커서를 왼쪽으로 한 칸 옮김 (커서가 문장의 맨 앞이면 무시됨)
D커서를 오른쪽으로 한 칸 옮김 (커서가 문장의 맨 뒤이면 무시됨)
B커서 왼쪽에 있는 문자를 삭제함 (커서가 문장의 맨 앞이면 무시됨) / 삭제로 인해 커서는 한 칸 왼쪽으로 이동한 것처럼 나타나지만, 실제로 커서의 오른쪽에 있던 문자는 그대로임
P $$라는 문자를 커서 왼쪽에 추가함

초기에 편집기에 입력되어 있는 문자열이 주어지고, 그 이후 입력한 명령어가 차례로 주어졌을 때, 모든 명령어를 수행하고 난 후 편집기에 입력되어 있는 문자열을 구하는 프로그램을 작성하시오. 단, 명령어가 수행되기 전에 커서는 문장의 맨 뒤에 위치하고 있다고 한다.

입력

첫째 줄에는 초기에 편집기에 입력되어 있는 문자열이 주어진다. 이 문자열은 길이가 N이고, 영어 소문자로만 이루어져 있으며, 길이는 100,000을 넘지 않는다. 둘째 줄에는 입력할 명령어의 개수를 나타내는 정수 M(1 ≤ M ≤ 500,000)이 주어진다. 셋째 줄부터 M개의 줄에 걸쳐 입력할 명령어가 순서대로 주어진다. 명령어는 위의 네 가지 중 하나의 형태로만 주어진다.

출력

첫째 줄에 모든 명령어를 수행하고 난 후 편집기에 입력되어 있는 문자열을 출력한다.

풀이

커서 왼쪽과 오른쪽을 따로 관리하면 명령 처리가 훨씬 단순해진다. 한쪽 문자열을 직접 중간 삽입하거나 삭제하면 느리지만, 왼쪽 스택과 오른쪽 스택으로 나누면 모든 명령을 거의 즉시 처리할 수 있다.

현재 코드는 leftStack, rightStack 두 덱으로 커서 양옆을 표현한다. L은 왼쪽 문자를 오른쪽으로 옮기고, D는 반대로 옮기며, B는 왼쪽에서 하나 제거하고 P는 왼쪽에 새 문자를 넣는다.

결국 커서를 움직인다는 행위를 두 스택 사이의 이동으로 바꾼 것이 핵심이다. 마지막에는 왼쪽에 남은 문자와 오른쪽에 남은 문자를 순서대로 합쳐 최종 문자열을 만든다.

코드

java
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayDeque;
import java.util.Deque;
 
public class Main {
 
    private static final StringBuilder sb = new StringBuilder();
    private static final BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
    public static void main(String[] args) throws IOException {
        String input = br.readLine();
        Deque<Character> leftStack = new ArrayDeque<>();
 
        for (char c : input.toCharArray()) {
            leftStack.push(c);
        }
 
        int m = Integer.parseInt(br.readLine());
        solve(leftStack, m);
 
        System.out.println(sb);
        br.close();
    }
 
    public static void solve(Deque<Character> leftStack, int m) throws IOException {
        Deque<Character> rightStack = new ArrayDeque<>();
 
        while(m-- > 0) {
            String command = br.readLine();
            char c = command.charAt(0);
 
            switch (c) {
                case 'L':
                    if (!leftStack.isEmpty()) {
                        rightStack.push(leftStack.pop());
                    }
                    break;
                case 'D':
                    if (!rightStack.isEmpty()) {
                        leftStack.push(rightStack.pop());
                    }
                    break;
                case 'B':
                    if (!leftStack.isEmpty()) {
                        leftStack.pop();
                    }
                    break;
                case 'P':
                    leftStack.push(command.charAt(2));
                    break;
            }
        }
 
        while (!leftStack.isEmpty()) {
            rightStack.push(leftStack.pop());
        }
 
        while(!rightStack.isEmpty()) {
            sb.append(rightStack.pop());
        }
    }
}

복잡도

  • 시간 복잡도: 초기 문자열과 명령을 리스트 연산으로 한 번씩 처리하므로 O(N+M)O(N + M)이다.
  • 공간 복잡도: 편집 중인 문자를 리스트에 저장하므로 O(N+M)O(N + M)이다.

마무리

커서를 두 스택 사이 경계로 보면 삽입과 삭제, 이동이 모두 가볍게 처리된다.