문자열 다루기 기본 / 프로그래머스, 파이썬 코딩테스트연습 연습문제

문제

문자열 s의 길이가 4 혹은 6이고, 숫자로만 구성돼있는지 확인해주는 함수, solution을 완성하세요. 예를 들어 s가 "a234"이면 False를 리턴하고 "1234"라면 True를 리턴하면 됩니다.

 

 

제출답안

def solution(s):
    len_s = len(s)
    answer = False
    if len_s == 4 or len_s ==6:
        try:
            list(map(int, [l for l in s]))
            answer = True
        except:
            pass
    return answer

 

 

제출 후 개선답안

in을 이용하여 판단코드 단순화
isdigit()으로 숫자판별 프로세스 단순화

한 줄로 적으려다가 길이가 4, 6이 아닌 경우
더 빨리 코드를 끝내고자 if문을 이중으로 적음

def solution(s):
    answer = False
    if len(s) in (4, 6):
        if s.isdigit(): answer = True
    
    return answer

 

 

비교결과 이중 if문과 속도차이가 없었음
(테스트 1~20까지 0.00ms인 경우가 각각 8건씩)

따라서 공간을 줄일 수 있는 한줄 코드가 더 유리한 것으로 결론

def solution(s):
    answer = len(s) in (4, 6) and s.isdigit()
    
    return answer

 

 

최초 코드 / if 이중 / 한줄코드

 

문제 바로가기