프로그래머스_짝지어 제거하기(JAVA)

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

코딩테스트 연습 - 짝지어 제거하기

문제 설명

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

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

b aa baa → bb aa → aa →

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

제한사항

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

---

입출력 예

s	result
baabaa	1
cdcd	0

입출력 예 설명

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

 

[문제 분석]

while문을 돌려 문자열이 모두 제거될 때까지 또는 문자열 내의 모든 문자를 check하면 O(n^2)으로 정확성 테스트는 통과하지만 효율성 테스트 부분에서 시간초과가 나게 된다.

그래서 stack을 이용하여 문제를 해결하였다. 

우선 Character를 저장할 stack을 선언한 뒤, String s를 구성하는 문자를 하나씩 저장할 것인데, 만약 stack 가장 최상위의 문자가 현재 저장할 문자와 같다면, 해당 문자를 push하지 않고 최상위 문자 또한 pop하여 제거하였다.

문자열을 이루는 마지막 문자까지 해당 작업이 완료되고 나서 stack의 크기가 0이면 모두 제거된 것이므로 1을 반환하고, 그렇지 않다면 제거되지 않은 문자가 존재하는 것이므로 0을 반환하도록 하였다.

 

 

 

[시간 초과 코드]

class Solution
{
    public int solution(String s)
    {
        int answer = 0;
        int i=0;
        while(true){
            if(s.length()==0){
                answer=1;
                break;
            }
            if(i==s.length()-1)
                break;
            if(s.charAt(i)==s.charAt(i+1)){
                if(i<s.length()-2)
                    s=s.substring(0,i)+s.substring(i+2);
                else
                    s=s.substring(0,i);
                i=0;
            } 
            else i++;
        }
        return answer;
    }
}

 

[최종 코드]

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