백준 5397번 Pseudo code (리뷰필요)

문제 링크

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

5397번: 키로거

 

시간 제한 / 메모리 제한

1 초

256 MB

 

문제

창영이는 강산이의 비밀번호를 훔치기 위해서 강산이가 사용하는 컴퓨터에 키로거를 설치했다. 며칠을 기다린 끝에 창영이는 강산이가 비밀번호 창에 입력하는 글자를 얻어냈다.

키로거는 사용자가 키보드를 누른 명령을 모두 기록한다. 따라서, 강산이가 비밀번호를 입력할 때, 화살표나 백스페이스를 입력해도 정확한 비밀번호를 알아낼 수 있다. 

강산이가 비밀번호 창에서 입력한 키가 주어졌을 때, 강산이의 비밀번호를 알아내는 프로그램을 작성하시오. 강산이는 키보드로 입력한 키는 알파벳 대문자, 소문자, 숫자, 백스페이스, 화살표이다.

 

입력

첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스는 한줄로 이루어져 있고, 강산이가 입력한 순서대로 길이가 L인 문자열이 주어진다. (1 ≤ L ≤ 1,000,000) 강산이가 백스페이스를 입력했다면, '-'가 주어진다. 이때 커서의 바로 앞에 글자가 존재한다면, 그 글자를 지운다. 화살표의 입력은 '<'와 '>'로 주어진다. 이때는 커서의 위치를 움직일 수 있다면, 왼쪽 또는 오른쪽으로 1만큼 움직인다. 나머지 문자는 비밀번호의 일부이다. 물론, 나중에 백스페이스를 통해서 지울 수는 있다. 만약 커서의 위치가 줄의 마지막이 아니라면, 커서 및 커서 오른쪽에 있는 모든 문자는 오른쪽으로 한 칸 이동한다.

출력

각 테스트 케이스에 대해서, 강산이의 비밀번호를 출력한다. 비밀번호의 길이는 항상 0보다 크다.

예제 입력

2
<<BP<A>>Cd-
ThIsIsS3Cr3t

예제 출력

BAPC
ThIsIsS3Cr3t

 

문제 풀이를 위해 생각한 것

1. 문제의 입력이 최대 1,000,000으로, 시간복잡도를 고려해야 풀 수 있다(O(N^2)이면 시간초과 가능성 다분함).
2. 배열을 활용해서 인덱스로 접근할까? -> 삽입 삭제가 빈번하기에 O(N) 연산은 지양해야 함
3. 커서를 포인터라고 생각하고, 삽입, 삭제 연산의 시간복잡도가 O(1)인 연결리스트를 활용해 풀 수 있을 것 같다.
4. 파이썬의 경우 연결리스트 문제를 안 풀어봤지만, pseudo code로는 가능할 것 같다.

 

사용한 자료구조

연결리스트.

삽입, 삭제가 빈번하고, 커서의 이동이 빈번하기에 이 연산들에 대해 O(1)의 시간복잡도를 가지는 연결리스트를 활용한다.

 

풀이 코드

각 테스트케이스 t에 대하여:
	linked = 연결리스트 선언 및 null값으로 초기화, cur, first, last = 첫 노드를 참조
	입력 word의 각 글자 s에 대하여:
		if s == 글자:
			if cur == null:
				cur = s
			elif cur != null:
				new_node 선언
				cur = new_node의 주소
				cur에 s 저장
		elif s == <:
			if cur == first:
				continue
			else:
				cur = prev_node
		elif s == >:
			if cur == last:
				continue
			else:
				cur = next_node
		elif s == -:
        	if cur != null:
				cur = prev_node
				삭제할 노드 free

 

시간 복잡도 분석

연결리스트의 각 삽입, 삭제 연산은 O(1)으로, 최대 N개의 글자 동안 O(N)의 시간 복잡도를 가짐.

 

문제에서 중요한 부분

입력의 크기가 1,000,000이기에, O(N)이내의 시간 복잡도로 알고리즘을 설계할 필요가 있었고, 이에 맞는 자료구조를 선택해야 했다.