본문 바로가기
코딩테스트 연습

프로그래머스 짝지어 제거하기 Level 2

by 동배_ 2021. 8. 5.

https://programmers.co.kr/learn/courses/30/lessons/12973?language=python3 

 

코딩테스트 연습 - 짝지어 제거하기

짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 2개 붙어 있는 짝을 찾습니다. 그다음, 그 둘을 제거한 뒤, 앞뒤로 문자열을 이어 붙

programmers.co.kr

문제 설명

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

 

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

b aa baa → bb aa → aa 

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

제한사항

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

처음 문제 해석 및 풀이 방법

1. while 문을 이용해 현재 문자열과 다음 문자열이 같다면 지금 문자열에서 제거한다(index 값을 따로줌)

2. index값이 문자열 크기를 초과하면 종료.

3. if문을 통해 문자열의 크기가 남아있으면 0 없으면 1을 반환한다. 

 

이후 문제 해석 및 풀이 방법

1. 스택을 이용해 이전 문자과 동일하지 않거나 없으면 문자을 하나씩 넣는다.

2. 넣은 문자가 현재 넣으려는 문자와 동일하면 스택을 pop한다.

3. 이를 for문을 통해 끝마친후 스택의 크기를 비교하여 남아있으면 0 없으면 1을 반환한다. 

내가 작성한 소스코드

def solution(s):
    stack = []
    for c in s:
        if stack:
            if stack[-1] == c:
                stack.pop()
            else:
                stack.append(c)
        else:
            stack.append(c)
    return 0 if stack else 1

위처럼 스택을 생성해 문자열을 하나씩 비교해 없거나 이전 문자열과 중복이 되지 않으면 push. 중복이 되면 pop을 하여 해결했다.

결론 및 고찰 

처음에는 while문을 통하여 현재 문자열과 다음 문자열이 같으면 pop을 이용해 문자열 내에 중복된 값을 2개씩 제거 했었다. 하지만 이렇게 하니 로직은 맞지만 효율성테스트에서 떨어져서 꽤나 고민했다. 결국 40분~50분을 고민하다가. 검색을 하게 됐고 스택을 이용해 시간복잡도를 줄일 수 있는 것을 알았다. 그냥 그 리스트에서 연속된 문자를 찾는 것 보다 훨씬 괜찮은 방법이였다. 문제를 보자말자 10초만에 풀 수 있을 것 같았는데 정말 아쉽다. 

 

댓글