[Cos Pro 1급][Python][기출문제] 1차 문제 3

문제 유형 : 빈칸 채우기, 함수

💠문제설명

문자열 형태의 식을 계산하려 합니다. 식은 2개의 자연수와 1개의 연산자('+', '-', '*' 중 하나)로 이루어져 있습니다. 예를 들어 주어진 식이 "123+12"라면 이를 계산한 결과는 135입니다.

문자열로 이루어진 식을 계산하기 위해 다음과 같이 간단히 프로그램 구조를 작성했습니다.

~~~
1단계. 주어진 식에서 연산자의 위치를 찾습니다.
2단계. 연산자의 앞과 뒤에 있는 문자열을 각각 숫자로 변환합니다.
3단계. 주어진 연산자에 맞게 연산을 수행합니다.
~~~

문자열 형태의 식 expression이 매개변수로 주어질 때, 식을 계산한 결과를 return 하도록 solution 함수를 작성하려 합니다. 위 구조를 참고하여 코드가 올바르게 동작할 수 있도록 빈칸에 주어진 func_a, func_b, func_c 함수와 매개변수를 알맞게 채워주세요.

💠매개변수 설명

문자열 형태의 식 expression이 solution 함수의 매개변수로 주어집니다.
* expression은 연산자 1개와 숫자 2개가 결합한 형태입니다.
* 연산자는 '+', '-', '*'만 사용됩니다.
* 숫자는 1 이상 10,000 이하의 자연수입니다.

💠Return 값 설명

expression을 계산한 결과를 return 해주세요.
* 계산 결과는 문자열로 변환하지 않아도 됩니다.

💠예시

| expression | return |
|------------|--------|
| "123+12"   | 135    |
'+'를 기준으로 앞의 숫자는 123이고 뒤의 숫자는 12이므로 두 숫자를 더하면 135가 됩니다.

💠코드

1단계. 주어진 식에서 연산자의 위치를 찾습니다.

def func_b(exp):
    for index, value in enumerate(exp):
        if value == '+' or value == '-' or value == '*':
            return index

• enumerate() 함수 사용

2단계. 연산자의 앞과 뒤에 있는 문자열을 각각 숫자로 변환합니다.

def func_c(exp, idx):
    numA = int(exp[:idx])
    numB = int(exp[idx + 1:])
    return numA, numB

• int()로 문자열을 숫자로 변환
• idx 위치 계산 조심하기

3단계. 주어진 연산자에 맞게 연산을 수행합니다.

def solution(expression):
    exp_index = func_b(expression)
    first_num, second_num = func_c(expression, exp_index)
    result = func_a(first_num, second_num, expression[exp_index])
    return result

• 마지막에 index 값을 넣는 게 아니라, 연산자 그 자체를 넣는 것 조심하기!
• tuple을 받아서 첫번째 숫자와 두번째 숫자로 unpack 해준다.

 

전체 코드

def func_a(numA, numB, exp):
    if exp == '+':
        return numA + numB
    elif exp == '-':
        return numA - numB
    elif exp == '*':
        return numA * numB
 # 사실 return이 있는데 elif가 필요했을까
    
def func_b(exp):
    for index, value in enumerate(exp):
        if value == '+' or value == '-' or value == '*':
#       if value in ('+', '-', '*')
            return index
        
def func_c(exp, idx):
    numA = int(exp[:idx])
    numB = int(exp[idx + 1:])
    return numA, numB

def solution(expression):
    exp_index = func_b(expression)
    first_num, second_num = func_c(expression, exp_index)
    result = func_a(first_num, second_num, expression[exp_index])
    return result

#The following is code to output testcase.
expression = "123+12"
ret = solution(expression)

#Press Run button to receive output. 
print("Solution: return value of the function is", ret, ".")

 

💠개인정리

• enumerate() 함수 사용
• int()로 문자열을 숫자로 변환

💠Python 답게 문제 풀기

expression = "123+12"
print(eval(expression))

# 결과
135

• eval() : 문자열 형태로 된 식을 주면, 그것에 대한 결과를 구해서 반환해준다.