[프로그래머스]LV. 2 - 짝지어 제거하기 (Python)

[프로그래머스]LV. 2 - 짝지어 제거하기(Python)

https://school.programmers.co.kr/learn/courses/30/lessons/12973#

문제 설명

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

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

b aa baa → bb aa → aa →

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

제한 조건

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

입출력 예

풀이과정

1. 주어진 문자열 s의 길이가 0이 될 때까지 같은 알파벳이 2개 붙어 있는 짝을 제거해주기만 하면 되는 문제이다.

• s의 길이를 0으로 만들 수 있으면 1을 리턴, 없으면 0을 리턴한다.

2. 스택으로 활용할 빈 리스트 stack_s를 선언하고, for문으로 s의 모든 문자를 탐색해야 한다고 생각했다.
3. for문이 한 번 실행될 때마다 if-elif-else문이 실행될 수 있도록 코드를 작성했다.

• 만약 stack_s의 길이가 0이라면 stack_s에 string을 추가하고, 현재 stack_s의 마지막 인덱스에 위치한 값이 string과 같다면 pop()을 통해 stack_s의 마지막 인덱스에 위치한 값을 제거해준다. 두 상황 모두 아니라면 stack_s에 string을 추가해주는 방식으로 코드를 작성했다.

4. 스택에 중복으로 문자가 추가되는 것을 방지하기 위해 위와 같이 코드를 작성했고, 이 후 삼항 연산자를 사용해 출력값을 리턴해주었다.

• elif 문 때문에 s의 모든 문자열 제거가 가능하다면 결국 stack_s의 길이는 0이 된다. 그래서 stack_s의 길이가 0이라면 1을 리턴, 아니라면 0을 리턴시켜주었다.

코드

def solution(s):
    stack_s = []
    for string in s:
        if len(stack_s) == 0:
            stack_s.append(string)
        elif stack_s[-1] == string:
            stack_s.pop()
        else:
            stack_s.append(string)
    return 1 if len(stack_s) == 0 else 0

후기

프로그래머스 레벨 2의 문제였지만 문제를 잘 읽어보면 정말 쉽고 간단한 문제였다. 그래서 한 번에 코드를 작성하고 통과할 수 있었다.
주어진 테스트 케이스가 2개뿐이라 실행 전에 3개의 테스트 케이스를 직접 추가해 주었고, 실행 결과 모두 통과할 수 있었다.
if-elif-else 문을 더 줄일 방법을 생각했으나 이 구문을 사용하면 스택에 string을 굳이 추가하지 않고 넘어갈 수 있다 생각해 이대로 진행했다.
문제를 푼 이후 다른 사람들의 코드를 봤는데 대부분 나랑 비슷한 방법을 사용해서 문제를 풀었다.