[PYTHON] 백준 2501번 약수 구하기

<문제1>

​

어떤 자연수 p와 q가 있을 때, 만일 p를 q로 나누었을 때 나머지가 0이면 q는 p의 약수이다.

6을 예로 들면

6 ÷ 1 = 6 … 0

6 ÷ 2 = 3 … 0

6 ÷ 3 = 2 … 0

6 ÷ 4 = 1 … 2

6 ÷ 5 = 1 … 1

6 ÷ 6 = 1 … 0

그래서 6의 약수는 1, 2, 3, 6, 총 네 개이다.

두 개의 자연수 N과 K가 주어졌을 때, N의 약수들 중 K번째로 작은 수를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 K가 빈칸을 사이에 두고 주어진다. N은 1 이상 10,000 이하이다. K는 1 이상 N 이하이다.

출력

첫째 줄에 N의 약수들 중 K번째로 작은 수를 출력한다. 만일 N의 약수의 개수가 K개보다 적어서 K번째 약수가 존재하지 않을 경우에는 0을 출력하시오.

예제 입력 1

6 3

예제 출력 1

3

​

​

<코드>

arr=[]
N, K = map(int, input().split())
cnt = 0
for i in range(1, N + 1):
    if N % i == 0:
        arr.append(i)
    cnt += 1

if K > len(arr):
    print(0)
else:
    print(arr[K-1])

배열에 N % i == 0 이 되는 수를 순서대로 저장하고 K 값이 배열의 크기보다 큰 경우 0 출력하는 것만 신경써주면 어렵지 않은 문제입니다.

​

​

<문제2> 약수의 개수 구하기

​

N 값이 주어지면 N 값의 약수의 개수를 구하시오.

​

<코드>

import math
N = int(input())
cnt = 0
for i in range(1, int(math.sqrt(N)) + 1):
    if N == i * i:
        cnt += 1
        continue
    if N % i == 0:
        cnt += 2

print(cnt)

혹은

N = int(input())
cnt = 0
i = 1
while i * i < N:
    if N % i == 0:
        cnt += 2
    i += 1
if N == i * i:
    cnt += 1


print(cnt)

)

약수의 개수를 구할 때 약수는 제곱해서 N 이되는 수 전까지 대칭성을 띄기 때문에 1 ~ N 까지 구할 필요가 없고 root(N) 까지만 개수를 구해 X2 를 하고 제곱수의 경우는 총 개수에 + 1 을 해주면 됩니다.

​

1번 문제 출처 - https://www.acmicpc.net/problem/2501

불러오는 중입니다...