[프로그래머스Lv0] OX퀴즈 - 파이썬

 

프로그래머스

<문제접근법>

공백을 기준으로 인덱스를 나누면 0,2,4는 숫자이고 1,3은 수식이다.

따라서 인덱스1번이 '+','-'일때고 구분하고 조건문을 각각 수행한다.

0,2인덱스를 더했을 때와 나누었을 때 결과값의 "O","X"를 리스트에 추가해준다.

def solution(quiz):
    answer = []
    for i in quiz :
        n = i.split(" ")
        if n[1] == "+" :
            if int(n[0])+int(n[2]) == int(n[4]) :
                answer.append("O")
            else :
                answer.append("X")
        elif n[1] == "-" :
            if int(n[0])-int(n[2]) == int(n[4]) :
                answer.append("O")
            else :
                answer.append("X")     
    return answer