[1403] C++ : 에디터

https://www.acmicpc.net/problem/1406

1406번: 에디터

 

문제

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

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

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

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

 

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

입력

첫째 줄에는 초기에 편집기에 입력되어 있는 문자열이 주어진다. 이 문자열은 길이가 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

 

풀이

처음에 이 문제를 너무 간단하게 생각했다. 그냥 string의 substr기능을 이용해 빠진 문자열을 자르고 붙였다. 또 붙일 때도 자르고 넣어서 붙였다. 그렇게 테스트도 엄청 잘 되고 가뿐하게 해결하나 싶었다. 

#include<iostream>
#include<string>

using namespace std;

int main()
{
	string s;
	string front, back;
	int n;
	cin >> s;
	getchar();
	cin >> n;
	int cursur = s.size();
	for (int i = 0; i < n; i++) {
		char command;
		cin >> command;
		switch (command)
		{
		case 'L':
			if(cursur != 0)
				cursur--;
			break;
		case 'D':
			if(cursur != s.size())
				cursur++;
			break;
		case 'B':
			if (cursur != 0) {
				front = s.substr(0, cursur - 1);
				back = s.substr(cursur, s.size());
				s = front + back;
				cursur--;
			}
			break;

		case 'P':
			char c;
			cin >> c;
			front = s.substr(0, cursur);
			back = s.substr(cursur, s.size());
			s = front + c + back;
			cursur++;			
		}


	}
	cout << s << '\n';
	return 0;
}

 

하지만, 결과는 시간초과가 나왔다. 여기서 시간초과가 나올만한게 substr과 붙이는 연산이었다. 그래서 string 헤더에 관련 내장함수가 있나 한번 찾아보았다.

놀랍게도 중간의 문자를 없애는 erase 와 중간에 문자를 추가하는 insert가 있었다. 그래서 그 기능을 사용해 코드를 수정해 주었다.

#include<iostream>
#include<string>

using namespace std;

int main()
{
	string s;
	string front, back;
	int n;
	cin >> s;
	getchar();
	cin >> n;
	int cursur = s.size();
	for (int i = 0; i < n; i++) {
		char command;
		cin >> command;
		switch (command)
		{
		case 'L':
			if(cursur != 0)
				cursur--;
			break;
		case 'D':
			if(cursur != s.size())
				cursur++;
			break;
		case 'B':
			if (cursur != 0) {
				s.erase(cursur-1, 1);
				cursur--;
			}
			break;

		case 'P':
			char c;
			cin >> c;
			s.insert(cursur, 1, c);
			cursur++;			
		}
	}
	cout << s << '\n';
	return 0;
}

하지만, 이것도 결국 시간초과의 문제가 생겼다. 결국 이 방법으로 안된다는 것을 깨닫고 새로운 자료구조부터 찾아야겠다고 생각했다. 가장 먼저 떠오른 것은 이중연결리스트(double linked list)였다. (예전에 써둔 글이 있으니 이중 연결리스트에 대해 잘 모르는 사람들은 읽어보길 바란다.)

2022.02.06 - [Programming/자료구조] - [자료구조] C - 연결리스트(linkedlist) - 3. 이중연결리스트

[자료구조] C - 연결리스트(linkedlist) - 3. 이중연결리스트

 

하지만, 들어온 char 하나씩 넣고 만들고 하는 과정도 너무 귀찮고, 더블 연결리스트 자체를 다루기가 귀찮았다. 그러다가 생각한 방법이 스택을 두개 만들어서 쪼개면 어떨까 하는 것이었다. 

빨간색 선이 커서라고 생각하면 된다.

글로 표현하는게 조금 어려워서 그림으로 표현해보았다. 커서를 기준으로 front와 back 스택을 만들고 front에 미리 다 저장해둔다음 왼쪽으로 이동하면 back에 옮기고 오른쪽으로 이동하면 front에 옮기는 방식이다. 그리고 추가도 그냥 front에 값을 추가하고 삭제도 pop하면 된다.

하지만, 스택으로 구현하면 마지막에 출력할 때 앞에서부터 출력을 해야하니 front에 있는걸 back으로 다 옮겨야하는 번거로움이 있었다. 그래서 양방향으로 입출력을 모두 할 수 있는 덱으로 구현하였다.

 

<최종 코드>

#include<iostream>
#include<string>
#include<deque>

using namespace std;

int main()
{
	deque<char> front;
	deque<char> back;
	char s[100001] = { 0 };
	int n;
	cin >> s;
	getchar();
	cin >> n;
	for (int i = 0; i < 100001; i++)
	{
		if (s[i] == 0)
			break;
		front.push_back(s[i]);
	}

	for (int i = 0; i < n; i++) {
		char command;
		cin >> command;
		switch (command)
		{
		case 'L':
			if (front.empty())
				break;
			back.push_back(front.back());
			front.pop_back();
			break;
		case 'D':
			if (back.empty())
				break;
			front.push_back(back.back());
			back.pop_back();
			break;
		case 'B':
			if (!front.empty())
				front.pop_back();
			break;
		case 'P':
			char c;
			cin >> c;
			front.push_back(c);
		}
	}
	
	while (!front.empty())
	{
		cout << front.front();
		front.pop_front();
	}
	while (!back.empty())
	{
		cout << back.back();
		back.pop_back();
	}
	cout << '\n';
	return 0;
}

 

 

덱 두개로 깔끔하게 성공하였다!