본문 바로가기
Algorithm

[백준 1406번] 에디터 _ JAVA 풀이 (시간초과 해결)

by jungwon3004 2022. 1. 20.
728x90
반응형
시간 제한                                  메모리 제한         제출            정답               맞힌 사람       정답 비율
JAVA 2초           512 MB 60389 16720 11018 26.782%

 

문제

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

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

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

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

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

입력

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

출력

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

예제 입력 1 복사

abcd
3
P x
L
P y

예제 출력 1 복사

abcdyx

예제 입력 2 복사

abc
9
L
L
L
L
L
P x
L
B
P y

예제 출력 2 복사

yxabc

예제 입력 3 복사

dmih
11
B
B
P x
L
B
B
B
P y
D
D
P z

예제 출력 3 복사

yxz
 

<풀이1>

 

아무래도 풀이를 java로 진행하기 때문에 class를 따로 만들어서 진행하는 게 어떨까 생각해보고 문제를 풀었다.

여기서도 나름 시간 조절을 위해 신경을 썼다.

빠른 입력과 처리를 위해 Scanner 대신에 BufferedReader를 사용했고,

빠른 수정을 위해 ArrayList 대신 LinkedList를 사용했다. 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Iterator;
import java.util.LinkedList;


public class Main {
	
	public static void main(String[] args) throws IOException, NumberFormatException {
		
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		
		String s = bf.readLine();
		
		Editor edit = new Editor(s);

		int M = Integer.parseInt(bf.readLine());

		for(int i=0;i<M;i++) {
			String comdTmp = bf.readLine();
			edit.commend(comdTmp);
		}
        
		System.out.println(edit.show());
	}	
}

class Editor{
	
	LinkedList<String> str;
	int len;
	int cur;
	
	// String s를 받아서 각각을 잘라서 String[] str에 담음 
	public Editor(String s) {
		str = new LinkedList<String>(Arrays.asList(s.split("")));
		len=str.size();
		cur=len;
	}
	
	public void commend(String comd) {
		String[] comdArr = comd.split(" ");

		switch(comdArr[0]){
		case "L":
			L();
			break;
		case "D":
			D();
			break;
		case "B":
			B();
			break;
		case "P":
			P(comdArr[1]);
			break;
		}
		
	}
	
	public String show() {
		String result="";
		
		Iterator<String> iterator = str.iterator();
		while(iterator.hasNext()) {
			result+=iterator.next();
		}
		
		return result;
	}
	
	private void L() {
		if(cur>0)
			cur--;
	}
	
	private void D() {
		if(cur<len) {
			cur++;
		}
	}
	
	private void B() {
		if(cur>0) {
			str.remove(cur-1);
			cur--;
			len--;
		}
	}
	
	private void P(String addStr) {
		str.add(cur, addStr);
		len++;
		cur++;
	}
	
}

class를 나누고 역할을 분배해뒀기 때문에 코드 자체는 매우 간단하다.

하지만 시간초과.........

 

728x90

<풀이2>

class를 따로 구현하고 그 instance를 만들어서 사용하는 것보다 하나의 Main class 안에 코드를 작성하는게 빠르다.

그래서 별도의 class를 없애고 Main class 안에 다 넣어보았다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.LinkedList;


public class Main {
	
	public static void main(String[] args) throws IOException, NumberFormatException {
		
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		
		// 첫 입력값으로 문자열을 받아줌
		String s = bf.readLine();
		
		LinkedList<String> str = new LinkedList<String>(Arrays.asList(s.split("")));
		int len=str.size();
		int cur=len;
		
		// 명령어의 개수 M을 받아줌
		int M = Integer.parseInt(bf.readLine());
		
		// 명령어를 받고 실행해줌
		for(int i=0;i<M;i++) {
			String comd = bf.readLine();
			
			String[] comdArr = comd.split(" ");
			
			switch(comdArr[0]){
			case "L":
				if(cur>0)
					cur--;
				break;
			case "D":
				if(cur<len) 
					cur++;
				break;
			case "B":
				if(cur>0) {
					str.remove(cur-1);
					cur--;
					len--;
				}
				break;
			case "P":
				str.add(cur, comdArr[1]);
				len++;
				cur++;
				break;
			}
		}

		for(String tmp:str)
			System.out.print(tmp);
		
	}	
}

아쉽게도 시간초과,.....,,,

 

찾아보니 이 두 풀이가 시간초과가 발생하는 가장 큰 이유는 "LinkedList"를 사용했기 때문이라고 한다.

 

물론 ArrayList와 LinkedList를 비교했을 때, LinkedList가 데이터 수정에 유리하다.

하지만 치명적인 단점이 존재하는데, 바로 데이터를 읽을 때 오래걸린다는 것이다.

 

LinkedList에서 add()와 remove()를 하면 처음부터 전체를 다 읽고 추가, 삭제하는 거라고 한다.

즉, O(n)이 된다는 것이다.

 

* LinkedList는 head와 tail 정보만 가지고 있기에 중간 data를 꺼내려면 처음부터 읽어나가야 한다.

 

반응형

<풀이3>

아무리 생각을 해봐도 더 좋은 방법이 떠오르지 않아서 결국 검색을 헀다. ㅠㅠ

 

[백준 - Java] 1406번 : 에디터

문제 더보기 https://www.acmicpc.net/problem/1406 한 줄로 된 간단한 에디터를 구현하려고 한다. 이 편집기는 영어 소문자만을 기록할 수 있는 편집기로, 최대 600,000글자까지 입력할 수 있다. 이 편집기에

minhamina.tistory.com

이 블로그에 있는 풀이를 참고했는데

와,,,,,,,,,,,,,,,,,, Stack을 이용해서 이걸 구현한다는 생각을 어떻게 했을까,,,,,,,,,,,,

진짜 세상에 똑똑한 사람은 많은가보다 ㅠㅠ

약간 원래 이런 풀이가 먼저 생각하긴 어려운데, 막상 보면 "아~ 그러네" 하고 고개를 끄떡이게 되는 풀이다 ㅠㅠ

 

 

아무튼 아이디어를 정리해보자.

 

- 문제 인식

위에서 <풀이1>과 <풀이2>에서 사용한 방법은 cur(커서)를 int 타입의 값으로 지정하고 그 값, 즉 그 커서의 주소를 변화시켜가면서 LinkedList에 접근을 해 가는 방법이었다.

물론 이 풀이가 절대 나쁘다는 건 아니다.

하지만 현재 이 풀이 때문에 "시간 초과"가 발생했다.

 

원인을 생각해보자.

이 문제의 원인은 LinkedList의 read 과정이다.

LinkedList의 특성상 read할 때 start인 0부터 순서대로 읽어야 한다.

index n인 값을 찾으려면 0부터 n까지 하나씩 체크해서 찾아야 한다.

즉, O(n)의 속도를 가진다는 것이다.

이보다 빠른 알고리즘을 찾아야 한다.

 

- 해결 방법

Stack의 시간복잡도는 O(1)이다.

따라서 이걸 활용해서 시간초롸를 해결해보자.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.EmptyStackException;
import java.util.Stack;


public class Main {
	
	public static void main(String[] args) throws IOException, NumberFormatException {
		
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		
		// stF (stack front), stB (stack back)
				Stack<String> stF = new Stack<>();
				Stack<String> stB = new Stack<>();
		
		// 첫 입력값으로 문자열을 받아줌
		String inStr = bf.readLine();
		
		// 그 inStr을 배열로 바꾸고 Stack stF에 넣어줌
		String[] inStrArr = inStr.split("");
		
		for(int i=0 ;i<inStrArr.length ;i++) {
			stF.add(inStrArr[i]);
		}
		
		// inStrArr은 이제 필요없으니 삭제
		inStr=null;
		inStrArr=null;
		
		// 명령어의 개수 M을 받아줌
		int M = Integer.parseInt(bf.readLine());
		
		// M번 명령어를 받고 실행해줌
		for(int i=0;i<M;i++) {
			// comd라는 명령어를 받음
			String comd = bf.readLine();
			String[] comdArr = comd.split(" ");
			
			switch(comdArr[0]){
			case "L":
				try {
					stB.push(stF.pop());
				}catch(EmptyStackException e) {}
				break;
			case "D":
				try {
					stF.push(stB.pop());
				}catch(EmptyStackException e) {}
				break;
			case "B":
				try {
					stF.pop();
				}catch(EmptyStackException e) {}
				break;
			case "P":
				stF.push(comdArr[1]);
				break;
			}
		}
		
		// 마지막에 출력을 해야하기 때문에 stB에 값을 다 몰아준 뒤 순서대로 pop() 해주면 된다.
		while(true) {
			try {
				stB.push(stF.pop());
			}catch(EmptyStackException e) {break;}
		}
		
		// String에 담아도 되지만,
		// 값을 하나씩 입력하고 추가하는 방식이기 때문에 String보다 StirngBuffer가 유리
		StringBuffer result = new StringBuffer();
		while(true) {
			try {
				result.append(stB.pop());
			}catch(EmptyStackException e) {break;}
		}
		
		System.out.println(result.toString());
	}	
}

 

728x90
반응형