짝지어 제거하기 (파이썬)

문제

링크

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

예를 들어, 문자열 S = baabaa 라면
b aa baa → bb aa → aa →
의 순서로 문자열을 모두 제거할 수 있으므로 1을 반환합니다.

제한사항

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

---

1. 정확성 PASS, 효율성 FAIL

풀이

1. 시간초과를 해결하기 위해 문자열 길이가 홀수이거나 알파벳의 갯수가 홀수인게 있으면 return 0을 했다.
2. .replace를 이용해서 짝이 있으면 ''로 변환했다.

코드

def solution(s):
    if len(s)%2:
        return 0
    
    cnt = [0]*26
    alp = list('abcdefghijklmnopqrstuvwxyz')
    dic = dict(zip(alp, cnt))
    for w in s:
        dic[w] += 1
    for v in dic.values():
        if v%2:
            return 0

    after = s
    while True:
        before = after
        for a in 'abcdefghijklmnopqrstuvwxyz':
            i = a*2
            after = after.replace(i, '')

        if after == before:
            return 0
        elif len(after) == 0:
            return 1

---

2. 정확성 PASS, 효율성 PASS

풀이

1. for와 while만으로 시간초과를 해결할 수 없음 -> stack 사용
2. stack이 비어있으면 채우기
3. stack이 존재하면 마지막과 같은지 비교하여 같으면 .pop(), 다르면 .append()
4. 마지막 최종 stack이 비어있으면 return 1, 아니면 return 0

코드

def solution(s):
    L = len(s)
    if L%2: return 0
    
    stack = []
    for i in range(L):
        if not stack:
            stack.append(s[i])
        elif s[i] == stack[-1]:
            stack.pop()
        else:
            stack.append(s[i])

    if len(stack): return 0
    else: return 1