코딩 테스트 in python

소수 구하기

바닷가쟤 2025. 4. 15. 14:10

코딩테스트를 알고리즘 별로 정리된 괜찮은 사이트를 발견했다.

코드트리인데 구독료가 굉장히 비싸다. . . 

아무튼 가볍게 풀고 있는데 문제도 많고 단계별로 풀기 좋은 것 같다.


함수를 이용한 합과 소수 판별

소수 찾는 문제가 계속 나오던데 이번에 확실히 외우고 넘어가기로 했다.

a, b = map(int, input().split())

def decimal(n):
    if n < 2:
        return False
    for i in range(2, int(n**0.5) + 1):
        if n % i == 0:
            return False
    return True

def decimalFunction(a, b):
    cnt = 0
    for i in range(a, b + 1):
        if decimal(i):
            if((int(i / 10) + int(i % 10)) % 2 == 0):
                cnt += 1
    return cnt

print(decimalFunction(a, b))

 

코드트리는 이런 식으로 기능을 함수로 빼게 유도하는 것 같았다.

함수형인 프로그래머스 + 입출력 받는 백준을 합친 느낌이다.


 

728x90