[문제 설명]
덧셈, 뺄셈 수식들이 '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합니다.
[나의 풀이]
answer = []
def solution(quiz):
for q in quiz:
s = q.split(' ')
a, b, c = int(s[0]), int(s[2]), int(s[4])
if ((s[1] == '-') & (a-b==c)) | ((s[1] == '+') & (a+b == c)):
answer.append("O")
else:
answer.append("X")
return answer
[설명]
1. split을 사용하여 공백을 기준으로 분리하였다. (숫자와 기호가 분리됨)
quiz = ["-3 - -4 = 1", "-2 + 7 = 51", "7 + 3 = 4"]
for q in quiz:
s = q.split(' ')
a, b, c = int(s[0]), int(s[2]), int(s[4])
print(s)
print(a, b, c)
print()
Out[1]:
['-3', '-', '-4', '=', '1']
-3 -4 1
['-2', '+', '7', '=', '51']
-2 7 51
['7', '+', '3', '=', '4']
7 3 4
2. append를 이용하여 "O" , "X" 를 모았다.
quiz = ["-3 - -4 = 1", "-2 + 7 = 51", "7 + 3 = 4"]
answer = []
for q in quiz:
s = q.split(' ')
a, b, c = int(s[0]), int(s[2]), int(s[4])
if ((s[1] == '-') & (a-b==c)) | ((s[1] == '+') & (a+b == c)):
answer.append("O")
else:
answer.append("X")
answer
Out[2]:
['O', 'X', 'X']
[다른 사람 풀이]
def valid(equation):
equation = equation.replace('=', '==')
return eval(equation)
def solution(equations):
return ["O" if valid(equation) else "X" for equation in equations]
[eval 함수]
- (expression)식을 문자열로 받아서, 실행하는 함수다.
for equation in quiz:
equation = equation.replace('=', '==')
print(eval(equation))
Out[3]:
True
False
False
- eval 함수 예
##### eval 함수 예 #####
## 문자열 ##
a = eval('"abc" + "def"')
print(a)
## 숫자 ##
b = eval("4 + 7")
print(b)
## abs ##
c = eval("abs(-15)")
print(c)
Out[4]:
abcdef
11
15
'Test > python' 카테고리의 다른 글
문자열 계산하기 (replace, split 사용) (0) | 2023.04.09 |
---|---|
배열의 유사도 (set) (0) | 2023.04.09 |
문자열안에 문자열 (find, in, not in) (0) | 2023.04.05 |
세균 증식 (비트 연산자: <<, >>, &, | ) (0) | 2023.04.04 |
문자열 정렬하기 (2) (sorted() 사용) (0) | 2023.04.04 |