문제 설명
짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 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을 반환합니다.
나의 풀이
첫번째 풀이
소스
from collections import deque
def solution(s):
stack = deque([])
for c in s:
stack.append(c)
if len(stack) >= 2 and stack[-1] == stack[-2]:
stack.pop()
stack.pop()
if stack:
return 0
else:
return 1
설명
- Stack을 이용하여서 쉽게 풀 수 있었다.
- stack을 이용해서 끝의 2개를 비교한 후 같으면 둘 다 pop합니다.
- 마지막에 stack이 비어있으면 1을 리턴합니다.
- list pop시에 O(n)이 소요된다고 하셔서 알려주신 deque를 이용하였습니다.
-
질문
- 문득 궁금해졌는데, 왜 list pop(0)이나 pop()이나 O(n)인가요? 어차피 시작과 끝에있는걸 제거하는 것이니 O(1)이 걸려야 되는게 아닌가요? 왜 O(n)으로 비효율적으로 제거를 하는걸까요?
결과
통과
리뷰
return 0 if stack else 1
로 줄일 수 있습니다. :)조건을 조금 바꿔서
stack.pop()
을 두 번하지 않도록 수정할 수 있습니다. :)for c in s: if stack and stack[-1] == c: stack.pop() else: stack.append(c)
deque
를 사용하지 않아도 괜찮습니다. :)답변
list의 pop(0)은 O(n)이지만 pop()은 O(1)입니다. :) 이렇게 구현된 이유는 python의 list가 성능을 위해 C언어의 pointer array로 구현되서 그렇습니다. pop(0)을 했을 때 값을 리턴하는 것은 O(1)이지만 각 요소들을 앞으로 한 칸씩 당겨줘야하기 때문입니다. 예를들어
[5, 1, 6, 2, 9, 8]
이라는 값이 있을 때1
을 제거하면[5, X, 6, 2, 9, 8]
상태가되고 각6, 2, 9, 8
을 한 칸씩 당겨줘야 합니다. 그렇기 때문에 0번째 값을 제거한다면 결과적으로 O(n)의 시간복잡도가 발생합니다. :)
스터디 리더의 풀이
def solution(s):
stack = []
for c in s:
if stack and stack[-1] == c:
stack.pop()
else:
stack.append(c)
return int(stack == [])