Lv.0 OX퀴즈

문제 설명

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

입출력

코드 풀이

def solution(quiz):
    results = []  # 수식의 결과를 저장할 리스트
    
    for equation in quiz:
        x, operator, y, _, z = equation.split()  # 수식을 요소로 분할
        
        # 수식 계산 및 결과 비교
        if operator == "+":
            result = int(x) + int(y)
        elif operator == "-":
            result = int(x) - int(y)
        
        # 결과와 주어진 Z 비교 후 "O" 또는 "X" 저장
        if result == int(z):
            results.append("O")
        else:
            results.append("X")
    
    return results

출처: 프로그래머스 코딩테스트 연습