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

문제 설명

문자열 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
  "aAb1B2cC34oOp"안의 자연수는 1, 2, 34 입니다. 따라서 1 + 2 + 34 = 37 을 return합니다.

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

---

내가 푼 정답 (정규표현식)

import re
def solution(my_string):
    answer = 0
    new_str = re.split(r"[^0-9]", my_string)
    for i in new_str:
        if i.isdigit():
            answer += int(i)
    return answer

정답 풀이1

def solution(my_string):
    s = ''.join(i if i.isdigit() else ' ' for i in my_string)
    return sum(int(i) for i in s.split())

정답 풀이2

import re

def solution(my_string):
    return sum([int(i) for i in re.findall(r'[0-9]+', my_string)])