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

문제 설명

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

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

b aa baa → bb aa → aa →

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

제한사항

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

오답 코드

import java.util.*;

class Solution {
    
    public int solution(String s) {
        
        int answer = 0;

        Set<String> doubleAlpha = new HashSet<>();
        // 해당 문자열에 포함된 알파벳 2개를 이어 붙인 StringSet 생성
        
        for(int i=0; i<s.length(); i++) {
            doubleAlpha.add(s.charAt(i)+""+s.charAt(i));
        }


        String str = s;
        while(true) {

            String origin = str;
            for(String doubles : doubleAlpha) {
                if(str.contains(doubles)) str = str.replaceAll(doubles, "");
                // doubles를 포함하고 있는 경우, 제거
                else continue;
            }
            if(str.equals("")) {
            	// 모두 제거된 경우 성공이므로 answer = 1로 변경하고 break
                answer = 1;
                break;
            }
            if(origin.equals(str)) break;
            // 변경 전, 후 결과 같은 경우 더이상 진전 없으므로 answer = 0으로 종료.
        }

        return answer;
    }
}

이렇게 했더니 효율성 로직에서 하나도 통과하지 못했다...ㅋㅋ

'질문하기' 눌러서 확인해보니, (O)n 시간 복잡도에 맞춰야지만 통과 한다는 힌트를 접수,

그 밑에 다른 댓글들을 통해 '스택'을 이용해서 풀어야 한다는 중요한 힌트를 얻었다.

 

정답 코드

import java.util.*;

class Solution {
    
    public int solution(String s) {
        
        Stack<Character> stack = new Stack<>();
        
        for(int i=0; i<s.length(); i++) {
            stack.push(s.charAt(i));
            // i번째 char PUSH
            
            if(stack.size() > 1) {
            // stack에 2개 이상 들어 있는 경우,
            
                char top = stack.pop();
                char secondTop = stack.peek();
                
                if(top == secondTop) stack.pop();
                // 1st Top과 2nd Top 값 같은 경우 모두 POP
                else stack.push(top);
                // 다른 경우, top 되돌리기
            }
        }

        if(stack.isEmpty()) return 1;
        // for문 실행 후, stack이 비어 있으면 가능하므로 1 리턴

        return 0;
        // stack 비어있지 않은 경우 불가능하므로 0 리턴
    }
}

이렇게 푸니 성공했다 !

 

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