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

문제 설명

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

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

b aa baa → bb aa → aa →

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

---

제한사항

• 문자열의 길이 : 1,000,000 이하의 자연수

---

입력, 출력

입/출력

---

풀이

def solution(s):
    answer = 0
    stack = []
    for i in s:
        if len(stack) == 0:
            stack.append(i)
        elif stack[-1] == i :
            stack.pop()
        else:
            stack.append(i)
    if len(stack) == 0:
        answer = 1
    else:
        answer = 0
    

    return answer

function solution(s)
{
    var answer = 0;
    let stack = []
    
    for(let i = 0; i<s.length; i++){
        if(stack.length === 0){
            stack.push(s[i])
        }else if(stack[stack.length -1] === s[i]){
            stack.pop()
        }else{
            stack.push(s[i])
        }
    }
    if(stack.length === 0) answer = 1
    else answer = 0
    
    return answer;
}

결과

---

코드 풀이

for문을 통해서 인덱스 하나하나를 접근하면서 삭제하면 효율성도 그렇고 시간 복잡도에서 너무 오래 걸려서 계속 틀렸다.

다른 사람들의 풀이를 보며 stack을 이용하는 것을 참고해서 풀었다.

1. 입력받을 string과 stack 선언
2. stack이 비어있다면 stack에 첫 글자를 append
3. 스택의 마지막 배열과 stack에서 다음 글자가 같다면 stack에 들어있는 배열과 같다는 것이므로 pop
4. 위에 조건에 해당하지 않다면 다른 글자이므로 stack에 append
5. 마지막 조건문을 통해서 성공했으면 1 실패했으면 0을 반환 (삼항 연산자를 이용할 수도 있음)