프로그래머스 / lv2 / 짝지어 제거하기

문제

짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 2개 붙어 있는 짝을 찾습니다. 그다음, 그 둘을 제거한 뒤, 앞뒤로 문자열을 이어 붙입니다. 이 과정을 반복해서 문자열을 모두 제거한다면 짝지어 제거하기가 종료됩니다. 문자열 S가 주어졌을 때, 짝지어 제거하기를 성공적으로 수행할 수 있는지 반환하는 함수를 완성해 주세요. 성공적으로 수행할 수 있으면 1을, 아닐 경우 0을 리턴해주면 됩니다.

예를 들어, 문자열 S = baabaa 라면

b aa baa → bb aa → aa →

의 순서로 문자열을 모두 제거할 수 있으므로 1을 반환합니다.

제한사항

• 문자열의 길이 : 1,000,000이하의 자연수
• 문자열은 모두 소문자로 이루어져 있습니다.

 

입출력 예

입출력 예 설명

입출력 예 #1
위의 예시와 같습니다.
입출력 예 #2
문자열이 남아있지만 짝지어 제거할 수 있는 문자열이 더 이상 존재하지 않기 때문에 0을 반환합니다.

 

해결

처음에는 StringBuffer의 메소드와 반복문으로 충분히 해결할 수 있다고 생각해서, 다소 복잡한 예외처리를 하면서 구현은 완료했다. 그러나, 효율성 테스트에서 실패해버려서 다른 사람의 코드를 보았다. 하.. 스택으로 풀면 되는 문제였다. 일단 두 가지 코드를 다 올리겠다.

 

코드

	class Solution
	{
	public int solution(String s)
	{
	StringBuffer str = new StringBuffer(s);
	int index = 0;
	// 길이가 0이 될 때 까지 반복
	while(str.length() > 0) {
	// 만약 index가 str의 길이만큼 된다면 false
	if(index == str.length() - 1)
	return 0;
	// 만약 다음 index와 같은 문자라면 둘 다 지우고 index를 한 칸 미룬다.
	if(str.charAt(index) == str.charAt(index+1)) {
	str.delete(index, index+2);
	if(index > 0)
	index--;
	}
	// 같지 않다면 한 칸 올린다.
	else
	index++;
	}
	//반복문을 나왔다면 true
	return 1;
	}
	}

view raw test.java hosted with ❤ by GitHub

이렇게 풀면 효율성이 떨어진다. stack을 이용하면 아주 간단하게 연산을 반복할 필요도 없다.

	import java.util.*;
	public class Solution {
	public static int solution(String s) {
	Stack<Character> stack = new Stack<>();
	// String s를 char array로 바꾸고 하나씩 뺀다
	for(char c : s.toCharArray())
	// 만약 stack이 안 비어있고 끝 요소와 같다면 pop
	if(!stack.isEmpty() && stack.peek() == c) stack.pop();
	// 그게 아니면 push
	else stack.push(c);
	// 마지막에 스택이 비어있다면 1, 아니면 0
	return stack.isEmpty() ? 1 : 0;
	}
	}

view raw test.java hosted with ❤ by GitHub

 

느낀점

항상 문제를 풀 때, 어떤 자료구조를 쓸 수 있는지 유심히 봐야겠다..

 

출처

https://lkhlkh23.tistory.com/148

[프로그래머스, 자바] 짝지어 제거하기 - 스택