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

문제 설명

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

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

b aa baa → bb aa → aa →

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

제한사항

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

입출력 예sresult

baabaa	1
cdcd	0

입출력 예 설명

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

문제 해결

문제를 읽었을 때 반복문을 통해 큰 어려움없이 해결할 수 있을것 같다고 생각했다. 하지만 제한사항에 문자열의 길이가 백만이하의 자연수라는걸 보고 일반적인 방법으로 풀면 시간초과를 피할 수 없고 최대한 O(n) 즉, for문 1번만을 이용해서 풀 수 있는 방법을 고민했고 Stack자료구조를 이용해서 쉽게 해결했다.

1. 주어진 문자열 s를 처음부터 끝까지 순회

2. 현재 문자열의 문자와 스택의 최상단에 있는 문자가 서로 일치하다면 스택을 pop

3. 그렇지 않다면 스택에 문자 push

4. for문 종료 시 만약 스택이 비어있다면 문자열을 모두 제거할 수 있으므로 1반환 그렇지 않다면 0

// 수도코드
Stack S
String str

for i -> str.size()
	if str[i] == S.top() -> S.pop()
     else
     	S.push(str[i]))
if S.empty -> return 1
return 0;

전체코드

#include <iostream>
#include <string>
#include <stack>

using namespace std;

typedef long long ll;

int solution(string s)
{
    int answer = 0;
    stack<char>S;
    
    for(ll i=0; i<s.size(); i++){
        if(!S.empty() and s[i] == S.top())
                S.pop();
        else
            S.push(s[i]);
    }
    if(S.empty())
        answer = 1;
 
    return answer;
}