백준 알고리즘 13420번 사칙연산(python)

문제의 제목이 사칙연산인 단순 사칙연산 문제이다.

for _ in range(int(input())):
    li = input().split()
    a, b, c = int(li[0]), int(li[2]), int(li[4])
    op = li[1]
    if op == '+':
        t = a+b
    elif op == '-':
        t = a-b
    elif op == '*':
        t = a*b
    elif op == '/':
        t = a//b
    print("correct" if t == c else "wrong answer")