본문 바로가기
알고리즘/프로그래머스

짝지어 제거하기

by Dr.섭도 2023. 11. 29.

문제

문제 설명

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

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

b aa baa → bb aa → aa →

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

제한 사항

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

입출력 예

입출력 예 설명

입출력 예 1
위의 예시와 같습니다.

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

결과

정확성 테스트

효율성 테스트

나의 풀이

첫 풀이

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

        String[] strArr = {
            "aa","bb","cc","dd","ee","ff","gg","hh","ii","jj","kk","ll","nn","mm","oo","pp","qq","rr",
            "ss","tt","uu","vv","ww","xx","yy","zz"};

        while(true) {
            int len = s.length();

            for(int i = 0; i < strArr.length; i++) {
                s = s.replaceAll(strArr[i], "");
            }
            if(len == s.length()) break;
        }

        if(s.length() == 0) return 1;
        return answer;
    }
}

replaceAll로 치환하려 했지만, 시간 초과로 채점이 불가능했다. 그래서 Stack을 이용하기로 했다.

두 번째 풀이

import java.util.*;

class Solution
{
    public int solution(String s)
    {
        int answer = 0;
        Stack<Character> stack = new Stack();
        char[] charArr = s.toCharArray();

        for(int i = 0; i < charArr.length; i++) {
            // stack이 비어있으면 그냥 넣으면 된다(빈 값 처리를 해줘야 한다)
            if(stack.isEmpty()) stack.push(charArr[i]);
            // stack의 전 단어가 동일하다면? 안넣어고 전 것도 뺀다
            else if(stack.peek() == charArr[i]) stack.pop();
            // 다른 경우에는 그냥 넣는다
            else stack.push(charArr[i]);
        }

        // stack이 비어있으면 모두 빠진 것이므로 가능한 상황
        if(stack.size() == 0) return 1; 
        return answer;
    }
}

Stack을 활용해 진행하면 O(n)의 시간 복잡도로 진행 가능하다

마무리

문자열을 replace로 치환하는 것도 좋지만, 여러번 진행해야 하는 경우에 많은 시간이 소요된다. 따라서 한 단어씩 앞 뒤로 비교하는 경우에는 Stack을 활용하도록 하자