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

문제

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

제한사항

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

입출력 예시

• 입출력 예 1
  • "aAb1B2cC34oOp"안의 자연수는 1, 2, 34 입니다. 따라서 1 + 2 + 34 = 37 을 return합니다.
• 입출력 예 2
  • "1a2b3c4d123Z"안의 자연수는 1, 2, 3, 4, 123 입니다. 따라서 1 + 2 + 3 + 4 + 123 = 133 을 return합니다.

코드 1

import re

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

코드 2

def solution(my_string):
    answer = 0
    temp = '0'
    for idx, string in enumerate(my_string):
        try:
            if string.isdigit():
                temp += string
                if my_string[idx+1].isdigit():
                    pass
                else:
                    answer += int(temp)
                    temp = '0'
        except:
            answer += int(temp)
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges