코딩 테스트/Programmers

lv.1 - 문자열 다루기 기본

worryback 2023. 8. 26. 01:09
728x90

문제 설명

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

 

제한 조건

  • s는 길이 1 이상, 길이 8 이하인 문자열입니다.
  • s는 영문 알파벳 대소문자 또는 0부터 9까지 숫자로 이루어져 있습니다.
s answer
"a234" false
"1234" true

solution.js

function solution(s) {
 if (s.length !== 4 && s.length !== 6) {
        return false;
    }

    return /^\d+$/.test(s);
}

isNaN이나, Number.isInteger를 써도 될거같은데, regex를 쓰는게 편해보이긴함.

 

solution.py

import re

def solution(s):
    if len(s) != 4 and len(s) != 6:
	    return False
    return bool(re.match(r'^\d+$', s))

 

728x90