전공/백준
1124
yha97
2023. 5. 11. 10:43
날짜 : 2023. 05. 11
사용 언어 : python
문제

코드
import sys
prime = [True] * 100001 # 소수면 True, 아니면 False
prime[0], prime[1] = False, False
for i in range(2, int(100001**0.5)+1):
if prime[i]:
idx = 2
while idx * i < 100001:
prime[idx*i] = False
idx += 1
def facto(n):
d = 2
cnt = 0
while n > 1:
#print(n)
if n % d == 0: # 나누어 떨어지면
n = int(n / d)
cnt += 1
else:
d += 1
if prime[cnt]:
return 1
else:
return 0
a, b = map(int, sys.stdin.readline().split())
res = 0
for i in range(a, b+1):
if prime[i]:
continue
res += facto(i)
print(res)
풀이
- 에라토스테네스의 체를 구현(최대값이 100,000이기 때문에 100,001까지 생성)
- 수를 입력받은 후 for문으로 탐색
- 입력받은 수가 소수일 때 / 아닐 때를 구분하면서 소인수분해 후 각 소인수의 개수에 따라 결과값을 카운트
알게된 점
- 풀었지만 TLE가 나왔고 그래서 다시 풀었다.
- 어떻게 해야 시간을 줄일까 고민한 결과 불필요하게 facto()함수에 보내지 말자는 아이디어가 떠올랐고, 소수는 무조건 소인수를 하나만 갖는다는 것이 생각났다.(ex) 17 = 1 * 17, 1은 소인수가 아님)
- 소수의 경우 소인수가 1개라는 것을 몰랐다.
- 소인수 : 주어진 자연수를 나누어 떨어뜨리는 약수 중에서 소수인 약수
참고 사이트
-