알고리즘 연습 [소수 찾기]

소수 찾기

문제

numberOfPrime 메소드는 정수 n을 매개변수로 입력받습니다.

1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하도록 numberOfPrime 메소드를 만들어 보세요.

소수는 1과 자기 자신으로만 나누어지는 수를 의미합니다.
(1은 소수가 아닙니다.)

10을 입력받았다면, 1부터 10 사이의 소수는 [2,3,5,7] 4개가 존재하므로 4를 반환
5를 입력받았다면, 1부터 5 사이의 소수는 [2,3,5] 3개가 존재하므로 3를 반환

나의 풀이

def numberOfPrime(n):
    # 1부터 n사이의 소수는 몇 개인가요?
    result = [i for i in range(2, n+1)]
    for num in range(2, n+1):
        for i in range(2, round(num**0.5) + 1):
            if num % i == 0 and num in result:
                result.remove(num)
    return len(result)

자신의 루트값까지만 나누어 떨어지지 않으면 소수임이 판별됨

다른 풀이

def numberOfPrime(n):
    def isprime(m):
        for i in range(3, round(m **0.5)+1, 2): 
            if(m % i is 0):
                return False
        return True
    return len([i for i in range(2, n+1) if(i is 2 or isprime(i) and i%2 is not 0)])