프로그래머스 Python 알고리즘 문제 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
• [연산자]는 + 와 - 중 하나입니다.

---

입출력 예quizresult

["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합니다.

 

 

문제를 풀어나가는 과정

1. 연산자 양 옆으로는 공백이 존재한다. 즉 split()메소드를 통해 공백과 연산자를 제거 해준다

def solution(quiz):
    answer = []
    for el in quiz:
        if ' - ' in el:
            a = el.split(' - ')
        if ' + ' in el:
            a = el.split(' + ')
            
    return answer

문자열을 자르려고 하니 문제!! 

' = ' 문자열도 구분자로 잘라줘야 하는데 split() 메소드를 사용할 때 여러 개의 구분자를 사용해 본 적이 없었다.

바로 구글링을 해보았다.

구글에 Python split 구분자 여러개 라고 검색해서 키워드를 찾을 수 있었다

re 라이브러리를 사용하면 구분자를 여러개로 사용할 수 있다는 것을 알게 되었고,

문제를 해결할 수 있게 되었다

re 사용법은 아래 블로그에서 확인하세요!

https://velog.io/@highgrace/%EA%B5%AC%EB%B6%84%EC%9E%90-%EC%97%AC%EB%9F%AC%EA%B0%9C%EB%A1%9C-string-to-list

나의 풀이

import re
def solution(quiz):
    answer = []
    for el in quiz:
        if ' - ' in el:
            num_split = re.split(' = | ',el)
            result = int(num_split[0]) - int(num_split[2])
            if result == int(num_split[3]):
                answer.append('O')
            else :
                answer.append('X')
        if ' + ' in el:
            num_split = re.split(' = | ',el)
            result = int(num_split[0]) + int(num_split[2])
            if result == int(num_split[3]):
                answer.append('O')
            else :
                answer.append('X')
            
    return answer