OX퀴즈

https://school.programmers.co.kr/learn/courses/30/lessons/120907

프로그래머스

---

문제 설명
덧셈, 뺄셈 수식들이 'X [연산자] Y = Z' 형태로 들어있는 문자열 배열 quiz가 매개변수로 주어집니다. 수식이 옳다면 "O"를 틀리다면 "X"를 순서대로 담은 배열을 return하도록 solution 함수를 완성해주세요.

---

제한사항

 

• 연산 기호와 숫자 사이는 항상 하나의 공백이 존재합니다. 단 음수를 표시하는 마이너스 기호와 숫자 사이에는 공백이 존재하지 않습니다.
• 1 ≤ quiz의 길이 ≤ 10
• X, Y, Z는 각각 0부터 9까지 숫자로 이루어진 정수를 의미하며, 각 숫자의 맨 앞에 마이너스 기호가 하나 있을 수 있고 이는 음수를 의미합니다.
• X, Y, Z는 0을 제외하고는 0으로 시작하지 않습니다.
• -10,000 ≤ X, Y ≤ 10,000
• -20,000 ≤ Z ≤ 20,000
• [연산자]는 + 와 - 중 하나입니다.

---

입출력 예

quiz	result
["3 - 4 = -3", "5 + 6 = 11"]	["X", "O"]
["19 - 6 = 13", "5 + 66 = 71", "5 - 15 = 63", "3 - 1 = 2"]	["O", "O", "X", "O"]

---

입출력 예 설명
입출력 예 #1

• 3 - 4 = -3 은 틀린 수식이므로 "X", 5 + 6 = 11 은 옳은 수식이므로 "O" 입니다. 따라서 ["X", "O"]를 return합니다.

입출력 예 #2

• 19 - 6 = 13 은 옳은 수식이므로 "O", 5 + 66 = 71 은 옳은 수식이므로 "O", 5 - 15 = 63 은 틀린 수식이므로 "X", 3 - 1 = 2는 옳은 수식이므로 "O" 따라서 ["O", "O", "X", "O"]를 return합니다.

---

주어진 문자열 quiz 안에 있는 수식들이 올바른지 아닌지를 확인하는 문제입니다.

 

이전에 사용했던 eval 함수가 생각나시나요? eval(expression)의 형태로 계산 결과를 반환하는 함수입니다.

여기서는 등호(=)를 기준으로 좌, 우가 나뉘어져 있는데, 좌측 식의 결과가 우측 값과 동일한지만 확인하면 간단하게 풀 수 있습니다.

파이썬으로 문제를 풀어보겠습니다.

def solution(quiz):
    answer = []
    # quiz에서 계산식 exp를 가져옵니다.
    for exp in quiz:
        # 좌측, 우측을 등호를 기준으로 분할합니다.
        l, r = exp.split("=")
        # 좌측의 계산결과가 우측의 값과 동일하다면
        if eval(l) == eval(r):
            # O을 추가합니다.
            answer.append("O")
        # 값이 동일하지 않다면
        else:
            # X를 추가합니다.
            answer.append("X")
    return answer

 

eval 함수를 사용해서 l과 r의 결과를 비교합니다. r에도 사용한 건 음수가 포함된 경우 때문입니다.

값이 같다면 O을, 다르면 X를 추가합니다.

 

이 코드를 한 줄로 표현하면 다음과 같습니다.

def solution(quiz):
    return ["O" if eval(l) == eval(r) else "X" for l, r in [exp.split("=") for exp in quiz]]

 

다만 보기는 편하지 않네요. 한 줄로도 표현할 수 있다는 걸 보여드리고 싶었습니다.