백준 / 스택 / 1406번 / 에디터 / C++

문제

한 줄로 된 간단한 에디터를 구현하려고 한다.

이 편집기는 영어 소문자만을 기록할 수 있는 편집기로, 최대 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

 

<리스트를 이용한 풀이>

#include <iostream>
#include <list>
using namespace std;

int main() {
	string str;
	cin >> str;

	list<char> editor(str.begin(), str.end());
	auto cursor = editor.end();
	// list<char>::iterator cursor = editor.end()와 같다.

	int n;
	cin >> n;

	for (int i = 0; i < n; i++) {
		char cmd;
		cin >> cmd;

		if (cmd == 'L') {
			if (cursor != editor.begin()) {
				cursor--;
			}
		}

		else if (cmd == 'D') {
			if (cursor != editor.end()) {
				cursor++;
			}
		}

		else if (cmd == 'B') {
			if (cursor != editor.begin()) {
				cursor--;
				cursor = editor.erase(cursor);
				// erase(iter) = iter가 가리키는 원소삭제, 
                		// 삭제한 원소의 다음 원소를 가리키는 iterator 반환
			}
		}

		if (cmd == 'P') {
			char input;
			cin >> input;

			editor.insert(cursor, input);
		}
	}

	for (const auto &element : editor) {
		cout << element;
	}

	return 0;
}

 

---

스택 문제이지만, 리스트가 더 편할거같아서 리스트로 일단 구현해보았다.

스택으로 하는법도 나중에 추가해야겠다.

 

auto를 이용해보았다.

일단 auto cursor = editor.end()를 보자.

원래 end() 메소드는 맨 마지막의 iterator를 반환해준다.

iterator는 일단 반복자라고 알아두자. (개념은 나중에 따로 포스팅 예정)

 

그리고 auto는 자동으로 형변환(?) 해준다고 생각하면 쉽다.(역시 개별 포스팅 예정)

auto는 자동으로 형변환을 해주는 간편함이 있지만, 대신 어떤 형인지 추정을 해야하기 때문에 초기화를 꼭 해줘야한다.

cursor라는 변수가 iterator이기 때문에 auto는 list<char>::iterator로 자동으로 변환해준다.

 

알고리즘은 어려울것이 없다. 문제 그대로 해주자.

 

마지막 for문에서 for(const auto &element : editor)가 등장하는데,

editor에 있는것에 다 접근해주면 된다고 생각하면 된다.

하지만 주소로 접근해주는 것이다.

만약 &element를 쓰지 않고, element를 쓰면 element에 값을 복사해주면서 접근해야 하기 때문에

시간이 오래 걸릴 수 있다.

const를 쓴이유는, 반복문안에서 값을 변경시키지 못하도록 하는것이다.

---

<스택을 이용한 풀이>

#include <iostream>
#include <stack>
using namespace std;

int main() {
	stack<char> l_stack;
	stack<char> r_stack;

	string str;
	cin >> str;

	int len = str.length();
	for (int i = 0; i < len; i++) {
		l_stack.push(str[i]);
	}

	int n;
	cin >> n;

	for (int i = 0; i < n; i++) {
		char cmd;
		cin >> cmd;

		if (cmd == 'L') {
			if (!l_stack.empty()) {
				r_stack.push(l_stack.top());
				l_stack.pop();
			}
		}

		else if (cmd == 'D') {
			if (!r_stack.empty()) {
				l_stack.push(r_stack.top());
				r_stack.pop();
			}
		}

		else if (cmd == 'B') {
			if (!l_stack.empty()) {
				l_stack.pop();
			}
		}

		else if (cmd == 'P') {
			char ch;
			cin >> ch;

			l_stack.push(ch);
		}
	}

	while (!l_stack.empty()) {
		r_stack.push(l_stack.top());
		l_stack.pop();
	}

	while (!r_stack.empty()) {
		cout << r_stack.top();
		r_stack.pop();
	}

	cout << "\n";

	return 0;
}

---

스택을 두개를 쓴다는 생각을 못했을땐, 리스트로 하는게 훨씬 간편하겠구나 생각했었다.

하지만 커서를 기준으로 커서 왼쪽의 스택과(l_stack) 커서 오른쪽의 스택(r_stack)을 사용하면 훨씬 쉬웠다.

 

알고리즘은 쉽지만 마지막에 출력하는거에서 헷갈릴수도 있겠다고 생각했다.

만약 커서를 왼쪽으로 옮기는 일이라고 하면, 왼쪽 스택의 top이 오른쪽 스택의 top으로 이동하는것이다.

이 행동을 위해선 왼쪽 스택의 top을 오른쪽 스택에 넣어주고, 왼쪽 스택을 pop하면 된다.

 

그리고 마지막에 출력해야 할때, 결국 한쪽으로 스택을 몰아주고 출력을 해줘야한다.

왼쪽 스택으로 몰아주게 되면, 먼저 들어온 문자가 스택의 맨밑에 있기 때문에 pop해서 하나씩 출력해줘야 하는데,

거꾸로 출력이된다.

 

따라서 간단하게 오른쪽 스택으로 스택을 몰아준후 pop한것을 차례대로 출력해주면된다.