본문 바로가기

알고리즘

코딩 테스트 연습 19일

728x90

코딩 테스트 연습 19일


프로그래머스, OX퀴즈

 

📄 나의 코드

quiz = ["19 - 6 = 13", "5 + 66 = 71", "5 - 15 = 63", "3 - 1 = 2"]

def solution(quiz):
    answer = []

    for str in quiz:
        arr_str = str.split(" ")

        if arr_str[1] == '-':
            result = int(arr_str[0]) - int(arr_str[2])
        elif arr_str[1] == '+':
            result = int(arr_str[0]) + int(arr_str[2])

        if result == int(arr_str[4]):
            answer.append("O")
        else:
            answer.append("X")
    return answer

print(solution(quiz))

 

📄 공부해볼 만한 코드1

def solution(quiz):
    answer = []
    for q in quiz:
        p, a = q.split("=")
        if eval(p) == int(a):
            answer.append("O")
        else:
            answer.append("X")
    return answer

 

📄 공부해볼 만한 코드2

def valid(equation):
    equation = equation.replace('=', '==')
    return eval(equation)

def solution(equations):
    return ["O" if valid(equation) else "X" for equation in equations]

'알고리즘' 카테고리의 다른 글

코딩 테스트 연습 21일  (0) 2022.12.21
코딩 테스트 연습 20일  (0) 2022.12.20
코딩 테스트 연습 15일&16일  (0) 2022.12.16
코딩 테스트 연습 14일  (0) 2022.12.14
코딩 테스트 연습 12일, 13일  (0) 2022.12.13