[프로그래머스] 소수 찾기

문제

문제 설명

1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하는 함수, solution을 만들어 보세요.

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

 

입력 & 출력

입출력 예

nresult

10	4
5	3

풀이 과정

1. 소수를 판별하는 함수를 만들어서 해결하려고 하였습니다. 소수 판별하는 함수는 총 3가지 정도로 많이 구성하지만 여기서는 에라토스테네스의 체의 개념을 사용한 어떤 숫자의 제곱근까지의 숫자까지만 판별하는 방식으로 시간 복잡도를 줄였습니다.

2. 1번 방식으로 소수를 구하는 함수를 구성한다면 그 결과값은 소수인지 아닌지 Boolean형태로 반환되게 됩니다. 이것을 조건문에 사용해서 소수일 경우에는 answer 변수를 1씩 증가시키고 이를 반환한다면 원하는 값을 얻을 수 있습니다.

 

코드

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


def solution(n):
    answer = 0
    for i in range(2, n + 1):
        if is_prime(i):
            answer += 1

    return answer