지원연구소 공식 블로그

[PROGRAMMERS] OX퀴즈 (LV. 0) - Python 본문

코딩테스트/프로그래머스

[PROGRAMMERS] OX퀴즈 (LV. 0) - Python

지원연구소 2024. 5. 2. 02:34

[ 문제 ]

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

 

 

[ 해결 방법 ]

1. 답을 입력 받을 리스트 answer 선언
2. quiz로 반복을 하면서 각 요소를 split() 메소드로 분리
3. 만약 연산자가 '+' 인데 피연산자의 합과 결과값이 같다면 'O', 다르다면 'X'를 answer에 추가
4. 연산자가 '+' 즉 '-' 인데 피연산자의 차와 결과값이 같다면 'O', 다르다면 'X'를 answer에 추가
5. 모든 반복이 끝난 후 answer 반환

 

 

 

[ solution.py ]

def solution(quiz):
    answer = []
    
    for i in quiz:
        op_str = i.split()
        if op_str[1] == '+':
            if int(op_str[0]) + int(op_str[2]) == int(op_str[4]):
                answer.append('O')
            else:
                answer.append('X')
        else:
            if int(op_str[0]) - int(op_str[2]) == int(op_str[4]):
                answer.append('O')
            else:
                answer.append('X')
    
    return answer

 

 

 

> 🌻피드백은 언제나 환영입니다🌻 <