[Programmers] 짝지어 제거하기

👀 문제 설명

문제

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

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

b aa baa → bb aa → aa →

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

 

제한사항

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

입출력 예

✍🏻풀이

s의 처음부터 두 글자씩 비교해가면서 빼려고 했다..! (많이 무식한 방법인듯..) 역시나 효율성 오류가 났고, 답도 틀렸다..! ㅎ

스택을 사용해서 푸는 문제였다. s에 한글자씩 접근해서 하는 건 동일하다.

가장 첫 글자를 먼저 스택에 넣는다. 이후의 문자들을 스택의 top과 비교하며 같다면 top에 있는 값을 빼주고, 다르다면 비교하는 문자를 스택에 push한다.

위 과정을 끝낸 후, 스택이 비어있다면 문자열을 제거할 수 있다는 뜻이므로 1을 반환하고, 아니라면 0을 반환한다.

 

코드

틀린 답

#include <stdio.h>
#include <iostream>
#include <iostream>
#include <string>

using namespace std;

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

    string origin = s;

    for (int i = 1; i < s.length(); i++) {
        if (s[i - 1] == s[i]) { // 같은 문자열일 경우
            s.erase(i - 1, i);
        }
        cout << s << endl;
    }

    if (origin == s) {
        answer = 0;
    }
    else {
        answer = 1;
    }

    return answer;
}

 

정답

#include <stdio.h>
#include <iostream>
#include <string>
#include <stack>

using namespace std;

int solution(string s)
{
    int answer = 0;
    stack<char> st;
    
    for (int i = 0; i < s.size(); i++) {
        // 스택이 비어있거나 이전에 스택 top의 값이 현재 s[i]와 다르면(중복이 아니면) push
        if (st.empty() || st.top() != s[i])
            st.push(s[i]);
        else // 스택 top의 값이 현재 s[i]와 같다면 top에 있는 값을 pop한다. (중복일 경우)
            st.pop();
    }
    
    // 스택이 비어있다면 문자열을 모두 제거할 수 있음 -> 1
    if (st.empty())
        return 1;

    return answer;
}