백준 BAEKJOON 13420번 사칙연산 [PYTHON/파이썬]

 

 

 

 

 

 

백준 BAEKJOON 13420번 사칙연산 [PYTHON/파이썬]

---

<문제 출처> (BRONZE Ⅱ)

https://www.acmicpc.net/problem/13420

13420번: 사칙연산

 

 

 

 

 

<풀이>

주어지는 식을 문자열로 받아서 리스트로 저장했다.

if문을 사용하여 +, -, *, / 마다의 조건을 건 뒤

결괏값과 비교하여 올바른 식이면 correct

그렇지 않으면 wrong answer를 출력했다.

 

 

 

 

 

 

 

<코드>

T = int(input())

for _ in range(T):
    expression = list(map(str, input().split()))

    if expression[1] == "+":
        result = int(expression[0]) + int(expression[2])
    elif expression[1] == "-":
        result = int(expression[0]) - int(expression[2])
    elif expression[1] == "*":
        result = int(expression[0]) * int(expression[2])
    elif expression[1] == "/":
        result = int(expression[0]) / int(expression[2])

    if result == int(expression[4]):
        print("correct")
    else:
        print("wrong answer")