본문 바로가기
Programmers/programmers

[프로그래머스]배열만들기 6 파이썬 // return a or b

by 딸기뚜왈기 2023. 12. 10.
def solution(arr):
    answer = []
    i = 0
    
    while i < len(arr):
        if len(answer) == 0:
            answer.append(arr[i])
            i += 1
        elif len(answer) > 0 and answer[-1] == arr[i]:
            answer.pop()
            i += 1
        elif len(answer) > 0 and answer[-1] != arr[i]:
            answer.append(arr[i])
            i += 1
    if answer == []:
        return [-1]
    else:
        return answer

 

 

 

def solution(arr):
    stk = []
    for i in range(len(arr)):
        if stk and stk[-1] == arr[i]:
            stk.pop()
        else:
            stk.append(arr[i])

    return stk or [-1]

 

return a or b 작동 방식 :

a 가 True로 평가되면 a를 반환

a 가 비어있거나 False로 평가되면 b를 반환