[프로그래머스] 짝지어 제거하기

유형 : 스택

 

문제

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

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

b aa baa → bb aa → aa →

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

 

제한사항

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

 

입출력 예

 

 

풀이

 

처음에는 단순히 문자열을 가지고 계속 뗐다 붙였다 하려고 했는데, 정확성과 효율성 테스트를 둘다 통과하지 못했다..ㅠ

효율성은 아예 0점이다.

 

 

알고보니 스택을 이용해 접근하면 훨씬 쉬운 문제였다.

 

aabbaa 가 있을때 

반복문을 차례로 돌면서

맨처음 stack이 비어있으므로 a push

그다음 문자가 a로 stack의 top과 동일하므로 pop

다음 b도 stack이 비어있으므로 b push

그다음 b는 stack의 top과 같으므로 pop...

 

이런식으로 연산을 하고 마지막에 stack에 아무것도 쌓여있는것이 없으면 1로 리턴하면 된다.

 

import java.util.Stack;
 
class Solution
{
    public int solution(String s)
    {
        int answer = -1;
        int len = s.length();
        Stack <Character> stack = new Stack<>();
        int start = 0;
        int end = 0;
        for(int i=0; i<s.length() ; i++){
            if(stack.size()==0)
                stack.push(s.charAt(i));
            else if(s.charAt(i)==stack.peek()){
                stack.pop();
            }
            else{
                stack.push(s.charAt(i));
            }
            
        }
        
        
        if(stack.size() == 0)
            answer = 1;
        else
            answer = 0;
        return answer;
    }
}

 

이렇게 하니 ALL PASS!!

 

 

문제링크 :

https://programmers.co.kr/learn/courses/30/lessons/12973#