자료구조 및 알고리즘/알고리즘

[알고리즘] 프로그래머스 2017 팁스타운 - 짝지어 제거하기

benjykim 2021. 1. 6. 17:41
반응형

프로그래머스 2017 팁스타운 - 짝지어 제거하기

url = https://programmers.co.kr/learn/courses/30/lessons/12973

문제 설명

짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 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을 반환합니다.


너무 쉬운 문제인데 문자열 자체에만 너무 집중해서 삽질한 문제.

처음 접근 방법

  • 문자열을 인덱스로 접근하여 s[idx]s[idx+1]이 같으면 s를 쪼개서 다시 process(s) 함수를 호출하는 방식이다. 문제 조건을 꼼꼼이 보지 않아 이런식으로 작성했다. 문제 조건을 유심히 살펴보는 습관이 필요하다.

  • 나의 풀이 방법을 설명하자면, "baabaa" 문자열의 경우 "aa" 문자열의 위치를 찾고 "aa" 이외의 문자열 "b" + "baa"을 다시 재귀적으로 처리하는 것이다.

  • 결국 모든 과정을 거치면 solution(s)함수 내 res변수에는 "" 값이 들어간다. "" 값이 아닌 경우엔 압축이 제대로 안되었다는 것이다.

  • 소스코드

    def solution(s):
        answer = 0
        res =''
        # s의 길이가 홀수인 경우 무조건 0 리턴
        if len(s) % 2 != 0:
            return answer
        # s의 길이가 짝수인 경우 
        else:
            res += (s)
            if not res:
                answer = 1
        return answer
    
    def process(s):
        result = ''
        isCompressed = False
    
        if not s:
            return result
    
        for i in range(len(s)-1):
            if s[i] == s[i+1]:
                result += process(s[:i] + s[i+2:])
                isCompressed = True
    
        if not isCompressed:
            result = s
        return result

스택을 이용한 풀이

  • 문자열의 길이 : 1,000,000 이하의 자연수라는 조건이 있기 때문에 O(n^2)이 되는 순간 못 푼다는 생각을 해야 한다. 따라서 스택을 사용하여 O(n)이 되도록 한다.

  • 소스코드

    def solution(s):
        answer = 0
        stack = []
        # s의 길이가 홀수인 경우 무조건 0 리턴
        if len(s) % 2 != 0:
            return answer
    
        for ch in s:
            if not stack:
                stack.append(ch)
                continue
    
            if stack[-1] == ch:
                stack.pop()
            else:
                stack.append(ch)
    
        if not stack:
            answer = 1
    
        return answer
반응형