짝지어 제거하기 [프로그래머스] (feat. stack..)

문제 풀이 정리

 

문제 설명

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

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

b aa baa → bb aa → aa →

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

제한사항

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

---

입출력 예

s / result

baabaa	1
cdcd	0

입출력 예 설명

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

 

코드

def solution(s):
    answer = 0
    
    if len(s) % 2 == 1: #문자열 길이가 홀수면 볼것도 없이 out
        answer = 0
        return answer
    
    alpha_list = []
    for alpha in s: # 문자열의 문자를 하나씩 보면 이전과 같다면 제거 (like stack)
        if len(alpha_list) == 0:
            alpha_list.append(alpha)
        elif alpha_list[len(alpha_list)-1] == alpha:
            alpha_list.pop()
        else:
            alpha_list.append(alpha)
    
    if len(alpha_list) == 0:
        answer = 1
    else:
        answer = 0
        
    return answer

 

문제 넋두리

 

문제 접근 자체를 잘못했었다.

문자열에서 특정 알파벳이 연속으로 나왔을때 (ex. aa가 나왔다)

해당 문자열을 전체 문장에서 지워버리자고 생각했다.

alpha_set = set(s)

for alpha in alpha_set:
	s = s.replace(alpha+alpha,"")

이러면 문제가 무엇인가 하니,,, abba의 경우 a는 이미 넘어 갔기 때문에 다시 보지 않는다.

 

좋아 두번째로 그럼 한번 검토하더라도 새로 생기는 문자열패턴이 있을수 있으니까

alpha_set = set(s)
bool = True

while bool:
    bool = False
    for alpha in alpha_set:
        s = s.replace(alpha+alpha,"")
        bool = True
        break

이렇게 한번 지울때마다 새로 생기는 문자열이 없는지 체크하기 위해 반복을 했었다.

그럼에도 시간초과가 났다. 당연하지...문자열이 길이가 기니까...

 

약속을 가기전 간단히 풀어야지 했던 문제라 마음이 급했다.

힌트를 봤더니

아... 이럴수가 이렇게 간단한 방법이 있는데 돌아가다니

게다가 스택을 이용하면 문자열의 길이만큼만 반복할수 있어서 위 방법보다 빠를거라 예상 할수 있었다.

삼차 코드

alpha_list = []
for alpha in s:
    if len(alpha_list) == 0:
        alpha_list.append(alpha)
    elif alpha_list[len(alpha_list)-1] == alpha:
        alpha_list = alpha_list[:-1]
    else:
        alpha_list.append(alpha)

틀릴리 없을꺼라 생각했다.

그런데

후...도저히 이유를 모르겠어서 구글링했다.

python stack이라고 검색했더니 한가지 함수를 발견했는데.

ref) ooeunz.tistory.com/7

[Python] Stack 사용하기

나처럼 alpha_list =alpha_list[:-1] 방법을 통한게 아닌

리스트 자체 내장 함수가 있었다.

 

그리고....

기쁜데...기쁘지 않다...

내시간....