[프로그래머스/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