[프로그래머스] 소수 만들기

소수 관련 문제를 풀다가 얻은 꿀팁

 

1. 조합 라이브러리

from itertools import combinations as cb

위의 라이브러리를 사용하면 알아서 조합을 구해준다.

이용 사례

nums 리스트에서 3개의 조합을 불러와 a로 저장하는 것을 볼 수 있다.

 

2. 소수 관련해서 구할 때 for - else 문을 이용하면 불필요한 계산을 안할 수 있다.

(내가 했던 방법)

1 ~ n 까지 for문을 돌려서 x%n==0 이면 count 증가, count가 3 이상이면 연산을 중단하고 count가 2면 소수로 판별

 

(더 나은 방법)

2 ~ n-1 까지 for문을 돌려서 x%n==0 이면 break, break 없이 진행된다면 소수로 판별

 

더 나은 방법을 이용한 사례