짝지어 제거하기

노누리·2022년 4월 21일
0

문제

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

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

b aa baa → bb aa → aa →

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

제한사항

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

입출력 예

sresult
baabaa1
cdcd0

입출력 예 설명

입출력 예 #1

위의 예시와 같습니다.

입출력 예 #2

문자열이 남아있지만 짝지어 제거할 수 있는 문자열이 더 이상 존재하지 않기 때문에 0을 반환합니다.

문제 풀이

고려한 점

  • 무조건 2개 붙어있으면 제거 -> 2개씩 제거

실패한 풀이

import re
def solution(s):
    idx=0
    for i in range(len(s)):
        #if s[idx]==s[idx+1]:
            #s=s.replace(s[idx]*2,'',1)
        #else:
            #idx+=1
        s=re.sub('(([a-z])\\2{1,})', '', s)
        
        if len(s)==0:
            return 1
        elif idx+1>=len(s):
            return 0
  • 정규표현식을 사용한 방식이 조금이나마 효율성이 증가했다.
  • replace() 사용시 효율이 떨어졌다.

stack을 이용하는 문제

def solution(s):
    answer = -1
    stack=[s[0]]

    for i in range(1,len(s)):
        if len(stack)!=0 and stack[-1]==s[i]:
            stack.pop(-1)
        else:
            stack.append(s[i])
    return 1 if len(stack)==0 else 0

위의 문제점을 해결하기 위해 stack을 이용해서 풀었다.

스택에 문자 하나씩 집어넣으면서 스택의 제일 위에 있는 값과 현재 가리키는 값이 같은 문자인 경우, 함께 제거해준다.
그렇지 않으면 스택에 문자를 넣어준다.
문자열의 끝에 도달할때까지 반복해주고 마지막까지 스택에 값이 남아있다면 모두 제거되지 못했기 때문에 return 0을 해준다.

다른 사람 풀이

def solution(s):
    answer = []
    for i in s:
        if not(answer):
            answer.append(i)
        else:
            if(answer[-1] == i):
                answer.pop()
            else:
                answer.append(i)    
    return not(answer)

stack을 이용하여 푼 풀이이다. 내가 푼 풀이와 조금씩 다르지만 스택을 이용했다는 점에서 같은 로직이라고 생각하면 될 것 같다.

풀고나면 같은 단어를 짝지어서 제거하는 간단한 풀이인데 stack에 익숙하지 않아서, 떠올리지 못해서 푸는데 시간이 걸렸던 문제이다.
stack을 무서워하지 말고 쓸 수 있는 상황에서 잘 사용해야겠다고 생각이 들었다.

profile
백엔드 개발자입니다.

0개의 댓글