[프로그래머스] LV.0 숨어있는 숫자의 덧셈(2)

문제

문자열 my_string이 매개변수로 주어집니다. my_string은 소문자, 대문자, 자연수로만 구성되어있습니다. my_string안의 자연수들의 합을 return하도록 solution 함수를 완성해주세요.

제한사항

- 1 ≤ my_string의 길이 ≤ 1,000
- 1 ≤ my_string 안의 자연수 ≤ 1000
- 연속된 수는 하나의 숫자로 간주합니다.
- 000123과 같이 0이 선행하는 경우는 없습니다.
- 문자열에 자연수가 없는 경우 0을 return 해주세요.

입출력 예시

my_string	result
"aAb1B2cC34oOp"	37
"1a2b3c4d123Z"	133

풀이

1. 연속되는 숫자를 1개의 자연수로 잘 처리하는 것이 중요하다.
2. 풀이 1에서는 모듈 없이 처리하는 방법이고 풀이 2는 모듈을 사용해 간단하게 처리하는 방법이다.

풀이 1

def solution(my_string):
    answer = 0
    temp = ''
    
    for i in my_string:
        if str.isdigit(i):
            temp+=i
        else:
            try:
                answer+=int(temp)
                temp = ''
            except:
                print("temp is empty")
    if temp != '':
        answer+=int(temp)
    return answer

• 문자가 자연수일 때 임시로 저장해둘 temp 문자열을 정의해준다
• 문자열을 for문으로 각 문자를 추출해서 if문으로 해당 문자가 숫자인지 str.isdigit()으로 판별해 준다.
• 만약 숫자일 때, temp에 더해줘서 연속되는 숫자를 찾을 수 있다.
• 해당 문자가 숫자가 아니면 숫자를 int()로 치환하여 answer에 더해주는데 try문을 사용하여 temp가 비어있는지 확인해준다.
• 모든 for문을 돌리고 temp가 비었는지 확인을 해주어 숫자가 남아있으면 answer에 더해준다. 이는 마지막 문자가 숫자일 시 더해지지 않는것을 방지해준다.

풀이 2

import re

def solution(my_string):
    answer = 0
    numbers = re.findall(r'\d+', my_string)
            
    for num in numbers:
        answer+=int(num)
    return answer

• re 모듈을 사용하기 위해 import 해준다.
• re.findall()로 모든 숫자를 추출하여 numbers 변수에 리스트로 저장을하는데 여기서 숫자가 연속되면 더해질 수 있도록 findall의 첫 번째 인자에서 r'\d+'를 해주도록 한다.
• 모든 numbers의 요소들을 answer에 더해준다.

오답노트

• 연속되는 숫자를 처리하는게 어려웠다.
• re모듈을 사용하면 쉽겠지만 사용을 안하고자 했는데, temp 변수를 설정하여 더해주는 방법이 있었다.