Algorithm/Programmers

[Programmers] Lv1. 소수 찾기

by somida 2021. 5. 15.

문제

바로가기

 

코딩테스트 연습 - 소수 찾기

1부터 입력받은 숫자 n 사이에 있는 소수의 개수를 반환하는 함수, solution을 만들어 보세요. 소수는 1과 자기 자신으로만 나누어지는 수를 의미합니다. (1은 소수가 아닙니다.) 제한 조건 n은 2이상

programmers.co.kr

 

 

코드

def solution(n):
    dp = [0, 0] + [1] * (n - 1)
    answer = 0
    for i in range(2, n + 1):
        if dp[i]:
            answer += 1
            # 소수가 맞다면, 그에 해당하는 배수들을 다 지운다.
            for j in range(2*i, n+1, i):
                dp[j] = 0
    return answer

 

 
반응형

댓글