[프로그래머스][문자열][C++] 짝지어 제거하기

짝지어 제거하기

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이하의 자연수
• 문자열은 모두 소문자로 이루어져 있습니다.

풀이 방법

처음엔 문자열이 짝지어져있으면 문자열을 지우는 방식으로 접근했는데 효율성 테스트에서 시간 초과가 발생했다.

그래서 질문하기를 살짝 보던 중 스택을 이용하면 된다길래 스택으로 접근했더니 쉽게 해결했다.

논리는 정말 간단하다. 문자열의 문자들을 차례대로 스택에 집어넣는 과정에서 스택의 top과 현재 넣을 문자가 같으면 문자를 넣지 않고 pop을 실행해주는 것이다.

이렇게 하면 결국 짝지어 모두 제거되면 스택은 empty 상태일 것이고, 문자가 하나라도 지워지지 않았다면 stack에 남아있을 것이다.

소스 코드

#include <iostream>
#include <string>
#include <stack>
using namespace std;

int solution(string s)
{
    int answer = 0;

    stack<char> st;

    st.push(s[0]);

    for(int i = 1; i < s.size(); i++) {
        if(st.empty())
            st.push(s[i]);
        else if(s[i] != st.top())
            st.push(s[i]);
        else
            st.pop();
    }

    if(st.empty())
        answer = 1;

    return answer;
}