🔗 문제 링크
https://www.acmicpc.net/problem/1406
❔ 문제 설명
한 줄로 된 간단한 에디터를 구현하려고 한다. 이 편집기는 영어 소문자만을 기록할 수 있는 편집기로, 최대 600,000글자까지 입력할 수 있다.
이 편집기에는 '커서'라는 것이 있는데, 커서는 문장의 맨 앞(첫 번째 문자의 왼쪽), 문장의 맨 뒤(마지막 문자의 오른쪽), 또는 문장 중간 임의의 곳(모든 연속된 두 문자 사이)에 위치할 수 있다. 즉 길이가 L인 문자열이 현재 편집기에 입력되어 있으면, 커서가 위치할 수 있는 곳은 L+1가지 경우가 있다.
이 편집기가 지원하는 명령어는 다음과 같다.
L 커서를 왼쪽으로 한 칸 옮김 (커서가 문장의 맨 앞이면 무시됨)
D 커서를 오른쪽으로 한 칸 옮김 (커서가 문장의 맨 뒤이면 무시됨)
B 커서 왼쪽에 있는 문자를 삭제함 (커서가 문장의 맨 앞이면 무시됨)
삭제로 인해 커서는 한 칸 왼쪽으로 이동한 것처럼 나타나지만, 실제로 커서의 오른쪽에 있던 문자는 그대로임
P $ $라는 문자를 커서 왼쪽에 추가함초기에 편집기에 입력되어 있는 문자열이 주어지고, 그 이후 입력한 명령어가 차례로 주어졌을 때, 모든 명령어를 수행하고 난 후 편집기에 입력되어 있는 문자열을 구하는 프로그램을 작성하시오. 단, 명령어가 수행되기 전에 커서는 문장의 맨 뒤에 위치하고 있다고 한다.
첫째 줄에 모든 명령어를 수행하고 난 후 편집기에 입력되어 있는 문자열을 출력한다.
⚠️ 제한사항
-
첫째 줄에는 초기에 편집기에 입력되어 있는 문자열이 주어진다.
-
이 문자열은 길이가 N이고, 영어 소문자로만 이루어져 있으며, 길이는 100,000을 넘지 않는다.
-
둘째 줄에는 입력할 명령어의 개수를 나타내는 정수 M(1 ≤ M ≤ 500,000)이 주어진다.
-
셋째 줄부터 M개의 줄에 걸쳐 입력할 명령어가 순서대로 주어진다.
-
명령어는 위의 네 가지 중 하나의 형태로만 주어진다.
💡 풀이 (언어 : Java)
처음 풀었던 풀이는 시간초과 판정을 받았다. StringBuilder에서 위치를 기록한 idx라는 변수를 이용해서 목표 위치에서 삭제, 삽입등을 하는 구조이다. 아마 for문에서 , 그리고 안의 if문에서 StringBuilder의 delete메소드와 insert 메서드는 삭제하고 삽입하고 하면서 다른 문자열들을 재정렬하므로 다시 으로, 총 의 시간복잡도가 걸리므로 탈락한 것 같다.
타인의 풀이를 보고 switch문 안에서 스택을 이용해 의 시간 복잡도가 걸리는 알고리즘을 참고했다. 왼쪽에는 기존 문자열을 넣어주고, 커서의 이동에 따라서 오른쪽 스택에 값을 넣고 빼는 구조이다. 마지막에는 왼쪽 스택의 값들을 오른쪽 스택에 넣어주고 순서대로 pop해서 출력해주면 된다. 하지만 이렇게 해도 시간초과 판정이 났다. ㅡㅡ; BufferedWriter를 이용해 출력의 시간까지 줄여주면 통과 판정을 받는다.
처음 풀이
public class Main {
private static StringBuilder strBuilder;
private static int idx;
private static void editor(String command) {
if (command.equals("L")) {
if (idx != 0) {
idx--;
}
} else if (command.equals("D")) {
if (idx != strBuilder.length()) {
idx++;
}
} else if (command.equals("B")) {
if (idx != 0) {
strBuilder.deleteCharAt(idx-1);
idx--;
}
} else {
strBuilder.insert(idx, command.charAt(2));
idx++;
}
}
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String str = br.readLine();
int n = Integer.parseInt(br.readLine());
strBuilder = new StringBuilder(str);
idx = strBuilder.length();
for (int i = 0; i < n; i++) {
editor(br.readLine());
}
System.out.println(strBuilder);
}
}고친 풀이
public class Main {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
String str = br.readLine();
int n = Integer.parseInt(br.readLine());
// 입력받은 문자열은 left스택에 넣어주고, 명령에 따라 right스택과 서로 주고 받는다.
Stack<String> rightStack = new Stack<>();
Stack<String> leftStack = new Stack<>();
for (String chr : str.split("")) {
leftStack.push(chr);
}
for (int i = 0; i < n; i++) {
String command = br.readLine();
switch (command) {
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;
default :
leftStack.push(String.valueOf(command.charAt(2)));
}
}
// 출력값은 left 스택의 맨 아래 원소부터이다. 그리고 right 스택의 원소와 합쳐야 하므로
// left 스택값들을 right 스택으로 옮겨주면 원래 출력할 문자열 순서가 된다.
while (!leftStack.isEmpty()) {
rightStack.push(leftStack.pop());
}
// right 스택 문자열을 위에서 부터 pop한다. 출력을 빠르게하기 위해 BufferedWriter 사용
while (!rightStack.isEmpty()) {
bw.write(rightStack.pop());
}
// 모두 출력하고 버퍼를 비움
bw.flush();
bw.close();
}
}