CS/알고리즘

백준 2581번 소수 파이썬 풀이

happy_life 2022. 2. 7. 11:20

문제

자연수 M과 N이 주어질 때 M이상 N이하의 자연수 중 소수인 것을 모두 골라 이들 소수의 합과 최솟값을 찾는 프로그램을 작성하시오.

예를 들어 M=60, N=100인 경우 60이상 100이하의 자연수 중 소수는 61, 67, 71, 73, 79, 83, 89, 97 총 8개가 있으므로, 이들 소수의 합은 620이고, 최솟값은 61이 된다.

입력

입력의 첫째 줄에 M이, 둘째 줄에 N이 주어진다.

M과 N은 10,000이하의 자연수이며, M은 N보다 작거나 같다.

출력

M이상 N이하의 자연수 중 소수인 것을 모두 찾아 첫째 줄에 그 합을, 둘째 줄에 그 중 최솟값을 출력한다. 

단, M이상 N이하의 자연수 중 소수가 없을 경우는 첫째 줄에 -1을 출력한다.

 

 

정답 코드

#2581소수

#소수 구하기 함수
def isPrime(n):
    if(n == 1):
        return False
    elif (n == 2):
        return True
    else:
        for i in range(2,int(n**0.5)+1):
            if n%i == 0:
             return False
        return True

all_list = list(range(2,10000))
prime_list = []

#for 문을 계속 돌리면 시간초과 나므로, 미리 소수를 배열에 넣어두기
for i in all_list:
    if isPrime(i):
        prime_list.append(i)

M = int(input())
N = int(input())

#배열의 인덱스, sum을 활용하는 아이디어 
specific_prime_list = []
for i in range(M,N+1):
    if(i in prime_list):
        specific_prime_list.append(i)

if len(specific_prime_list) == 0:
    print(-1)
else:
    print(sum(specific_prime_list))
    print(specific_prime_list[0])

 

 

 

 

시간초과되지 않도록 한번만 for문을 돌려 소수 배열을 만들었음

prime_list

 

이후 범위에 맞는 소수를 찾기 위해 또 하나의 소수 배열을 만들었음

specific_prime_list

 

 

범위 안에서 소수인 경우만 따로 체크해서 배열에 더해줌(append)

for i in range(M,N+1):
    if(i in prime_list):
        specific_prime_list.append(i)

 

다른 풀이와의 비교

 

M=int(input())
N=int(input())
arr=[]

for i in range(M,N+1):
    if i==1:
        pass
    elif i==2:
        arr.append(i)
    else:
        for j in range(2, i):
            if i%j==0:
                break
            elif j==i-1:
                arr.append(i)
if len(arr)==0:
    print('-1')
else:
    print(sum(arr))
    print(min(arr))

 

나의 풀이와 달리 따로 소수 구분 함수를 만들지 않았음. 소수 구분 + 범위 설정 두 가지의 사고방식으로 진행되는 나의 풀이에 비해 단순명료함. 

문제에서 주어진 조건이 시간이 많이 쓰이지 않는 부분이므로 굳이 배열로 우회해서 2번의 논리과정을 거칠 필요는 없었음. 따라서 상황에 따라서 코드를 줄일 수 있음을 명심해야겠음.