[백준] 13420 사칙연산

import sys
sys.stdin = open('1.txt')

n = int(input())
for i in range(n):
    li = input().split(' ')
    flag = False
    if li[1]=='*':
        tmp = int(li[0])*int(li[2])
        if tmp == int(''.join(li[-1:])):
            flag=True
    elif li[1]=='+':
        tmp = int(li[0])+int(li[2])
        if tmp == int(''.join(li[-1:])):
            flag=True
    elif li[1]=='-':
        tmp = int(li[0])-int(li[2])
        if tmp == int(''.join(li[-1:])):
            flag=True
    elif li[1]=='/':
        tmp = int(li[0])/int(li[2])
        if tmp == int(''.join(li[-1:])):
            flag=True

    if flag==True:
        print('correct')
    else:
        print('wrong answer')

 

다른사람 풀이

import sys;input()
for s in sys.stdin:
	a,b=s.split("=")
	print(["correct","wrong answer"][int(b)!=eval(a)])

- eval 이라는 함수는 문자열을 그대로 실행시켜 준다고 함.

 

다른 사람 풀이 2

import sys
def is_correct(s):
    tmp = s.split()
    for i in range(0,5,2):
        tmp[i] = int(tmp[i])
    if tmp[3]!='=':
        return False
    if tmp[1]=="*":
        if tmp[0]*tmp[2]==tmp[4]:
            return True
    elif tmp[1]=="+":
        if tmp[0]+tmp[2]==tmp[4]:
            return True
    elif tmp[1]=="-":
        if tmp[0]-tmp[2]==tmp[4]:
            return True
    elif tmp[1]=="/":
        if tmp[0]//tmp[2]==tmp[4]:
            return True
    return False
n = sys.stdin.readline()
for i in range(int(n)):
    s1 = sys.stdin.readline().rstrip()
    print("correct" if is_correct(s1) else "wrong answer")

배운점

input()보다는 sys.stdin.readline()을 사용하는 것이 속도 개선에 도움이 된다.

 

다른 사람 풀이 3

r=""
for _ in range(int(input())):
    s=input().split()
    if int(eval(s[0]+s[1]+s[2]))==int(s[-1]):
        r+="correct\n"
    else:
        r+="wrong answer\n"
print(r,end="")