[Programmers - lv01] 문자열 다루기 기본 (cpp / python)

문자열 다루기 기본

코딩테스트 연습 - 문자열 다루기 기본

문제 설명

길이가 4 또는 6인 문자열 중 숫자만 있는지 확인하는 문제.

문제 풀이

소스코드 : C++

#include <string>
#include <vector>

using namespace std;

bool solution(string s) {
    int len = s.size();
    if(len==4||len ==6)
        for(int i=0; i<s.size(); i++)
            if(s[i]>=95) return false;
    else return false;
    return true;
}

소스코드 : Python

len(s) in (4,6) 이 코드가 참 기발했다.
(4,6)을 가지고 있는 튜플 안에 len(s)와 같은 값이 있다면 True 그렇지 않다면 False 를 반환한다.

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