[프로그래머스/Python] OX퀴즈

문제

https://school.programmers.co.kr/learn/courses/30/lessons/120907

  • 덧셈, 뺄셈 수식들이 'X [연산자] Y = Z' 형태로 들어있는 문자열 배열 quiz가 매개변수로 주어집니다. 수식이 옳다면 "O"를 틀리다면 "X"를 순서대로 담은 배열을 return하도록 solution 함수를 완성해주세요.

풀이

quiz 내의 문제를 하나씩 루프문을 통해 접근한 뒤 문자열을 split하여 연산자에 따라 분기하여 처리하였다.

def solution(quiz):
    rst = []
    for q in quiz:
        [x,op,y,_,answer] = q.split()
        if op == "-":
            if int(x) - int(y) == int(answer):
                rst.append("O")
            else:
                rst.append("X")
        else:
            if int(x) + int(y) == int(answer):
                rst.append("O")
            else:
                rst.append("X")
    return rst