본문 바로가기

알고리즘&자료구조/프로그래머스

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

728x90

[문제]

짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다.

먼저 문자열에서 같은 알파벳이 2개 붙어 있는 짝을 찾습니다.

그다음, 그 둘을 제거한 뒤, 앞뒤로 문자열을 이어 붙입니다.

이 과정을 반복해서 문자열을 모두 제거한다면 짝지어 제거하기가 종료됩니다.

문자열 S가 주어졌을 때, 짝지어 제거하기를 성공적으로 수행할 수 있는지 반환하는 함수를 완성해 주세요.

성공적으로 수행할 수 있으면 1을, 아닐 경우 0을 리턴해주면 됩니다.

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

baabaa →bbaa →aa→

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

제한사항

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

[풀이]

빈 깡통리스트를 만들어서 거기에 담을 때마다 직전에 담긴 문자와 담으려는 문자가 같은지 비교하고,

같다면 리스트에 담긴 문자를 제거하고 다음 문자로 넘어가 다시 비교해주는 형태로 풀었다.

그렇게해서 만들어진 리스트가 여전히 깡통이면 모두 제거되었다는 뜻이므로 1을, 문자가 남아있다면 0을 반환.

 

def solution(s):
    list_s = [] # 깡통 만들기
    for i in range(len(s)):
        if i == 0: 
            list_s.append(s[i]) # 처음은 일단 넣기
        elif i > 0:
            if len(list_s) > 0:
                if s[i] == list_s[-1]:
                    list_s.pop() # 비교해서 같으면 빼기
                else:
                    list_s.append(s[i]) # 다르면 넣기
            else:
                list_s.append(s[i])
    if len(list_s) > 0:
        return 0 # 다 제거 되지 않으면 0
    else:
        return 1 # 다 제거되면 0