[BaekJoon] 13420번 : 사칙연산 (Python)

코딩 1일 1문제! 오늘의 문제는 사칙연산 입니다.

13420번: 사칙연산

Solution

def arithmetic_operation(mathematical_expression):
    is_correct = "wrong answer"
    
    num1, operator, num2, _, result = mathematical_expression.split()
    num1, num2, result = int(num1), int(num2), int(result)
    if (operator == "*") and (num1 * num2 == result):
        is_correct = "correct"
    elif (operator == "/") and (num1 / num2 == result):
        is_correct = "correct"
    elif (operator == "+") and (num1 + num2 == result):
        is_correct = "correct"
    elif (operator == "-") and (num1 - num2 == result):
        is_correct = "correct"
    
    return is_correct
        
if __name__ == "__main__":
    for _ in range(int(input())):
        mathematical_expression = input()
        print(arithmetic_operation(mathematical_expression))

Solution 풀이

먼저 입력 받은 수식을 공백으로 나눈 뒤

첫번째 세번째 숫자를 두번째 operator를 바탕으로 덧셈, 뺄셈, 곱셈, 나눗셈 한 값이 

다섯번째 값과 같다면 "correct"를 그렇지 않다면 "wrong answer"를 출력하도록 합니다.

SOMJANG/CODINGTEST_PRACTICE