프로그래머스 - OX퀴즈

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

프로그래머스

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

#제한사항
연산 기호와 숫자 사이는 항상 하나의 공백이 존재합니다. 
단 음수를 표시하는 마이너스 기호와 숫자 사이에는 공백이 존재하지 않습니다.
1 ≤ quiz의 길이 ≤ 10
X, Y, Z는 각각 0부터 9까지 숫자로 이루어진 정수를 의미하며,
각 숫자의 맨 앞에 마이너스 기호가 하나 있을 수 있고 이는 음수를 의미합니다.
X, Y, Z는 0을 제외하고는 0으로 시작하지 않습니다.
-10,000 ≤ X, Y ≤ 10,000
-20,000 ≤ Z ≤ 20,000
[연산자]는 + 와 - 중 하나입니다.

#입출력 예
quiz : ["3 - 4 = -3", "5 + 6 = 11"],	result : ["X", "O"]
quiz : ["19 - 6 = 13", "5 + 66 = 71", "5 - 15 = 63", "3 - 1 = 2"], result : ["O", "O", "X", "O"]

# 입출력 설명
(1) 
3 - 4 = -3 은 틀린 수식이므로 "X", 
5 + 6 = 11 은 옳은 수식이므로 "O" 입니다.
따라서 ["X", "O"]를 return합니다.

(2) 
19 - 6 = 13 은 옳은 수식이므로 "O", 
5 + 66 = 71 은 옳은 수식이므로 "O", 
5 - 15 = 63 은 틀린 수식이므로 "X", 
3 - 1 = 2는 옳은 수식이므로 "O" 
따라서 ["O", "O", "X", "O"]를 return합니다.

 

↓↓↓↓ 문제 풀이 ↓↓↓↓↓

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

input값으로 리스트가 들어올 때, 이것을 공백기준 나눠줘야 하는 상황.
하지만 리스트에 split()함수를 사용한다면 
AttributeError: 'list' object has no attribute 'split' => split은 str을 input으로 받기 때문에 AttributeError가 발생함 주의
quiz를 for문 순회해서 split해주고 각 연산자에 해당하는 연산조건을 걸어주었다.
문자열의 결괏값이랑 실제 연산결과가 일치하다면 "O" 또는 "X"를 answer리스트에 append하여 return하면 끝