[코딩 테스트] 프로그래머스_OX퀴즈(Python)

문제 설명

덧셈, 뺄셈 수식들이 '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합니다.

 

코드

def solution(quiz):
    answer = []
    for i in quiz:  # 첫번째 수식이 i에 저장...차례로 다음 수식이 저장
        tmp = i.split()
        if tmp[1] == '+':
            r = int(tmp[0]) + int(tmp[2])
        else:
            r = int(tmp[0]) - int(tmp[2])
            
        if r == int(tmp[4]):
            answer.append("O")
        else:
            answer.append("X")
            
    return answer

 

우선 비어있는 배열인 answer를 생성해줍니다. for 반복문을 이용해서 quiz안에 있는 첫번째 요소를 꺼내와서 i에 넣어줍니다. 예를 들어, quiz가 ["3 - 4 = -3", "5 + 6 = 11"]일 경우에 첫번째 반복이 돌 때 첫번째 요소인 "3 - 4 = -3"가 i에 들어가게 됩니다.

 

그 후 i의 요소를 split함수를 이용해서 문자열을 쪼개줍니다. 그리고 그것을 tmp에 넣어줍니다.

다시 말하자면, "3", "-", "4", "=", "-3" 이런 식으로 쪼개집니다.

 

그렇게 쪼개진 것을 가지고 if문을 이용하여 tmp의 첫번째 요소(tmp[1])가 "+"일 경우, tmp의 0번째 요소와 2번째 요소를 int형으로 변환해준 후 두 요소를 더해줍니다. 그리고 r에 저장해줍니다.

그게 아니라면, 두 요소를 빼준 후 r에 저장해줍니다.

 

다시 if문을 이용하여 위에서 저장해준 r이 int형으로 바꾼 tmp[4](tmp의 4번째 요소/즉, 수식의 결괏값)가 r과 같을 경우, answer에 append를 이용하여 "O"를 붙여줍니다.

그게 아니라면, "X"를 붙여줍니다.

 

마지막으로 answer를 return합니다.

 

끝!!