파이썬, 양의 정수 네자리 또는 여섯자리 검사

 

파이썬, 양의 정수 네자리 또는 여섯자리면 "YES", 아니면 "NO"를 출력하기.

글. 오상문 sualchi@daum.net

[입력] 양의 정수
[출력] "YES" or "NO"

 

def solution1(n):
    if n>=1000 and n<=9999:
        return "YES"
    if n>=100000 and n<=999999:
        return "YES"
    return "NO"
#------------------------------

def solution2(n):
    count=0
    while n>0:  # 0보다 큰 동안 반복
        count += 1
        n //= 10
    if count==4 or count==6:
        return "YES"
    return "NO"
#------------------------------

def solution3(n):
    s = str(n)
    if len(s)==4 or len(s)==6:
        return "YES"
    return "NO"
#------------------------------

num = int(input("정수: "))

print(solution1(num))
print(solution2(num))
print(solution3(num))

 

<이상>