[알고리즘]백준 2501번

김도연·2024년 1월 1일

알고리즘

목록 보기
2/56

어떤 자연수 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번째로 작은 수를 출력하는 프로그램을 작성하시오.
예제 입력 1 : 6 3
예제 출력 1 : 3

예제 입력 2 : 25 4
예제 출력 2 : 0

[내 코드]

N,K=map(int,input().split())
a=list()
for i in range (1,N+1):
	if(N%i==0):
		a.append(i)
if(len(a)<K):
	print(0)
else:
	print(a[K-1])

*K번째라는 워딩을 보고 먼저 리스트 형식으로 구현해야겠다고 생각

[해설코드]

N, K=map(int,input().split())
cnt=0
for i in range(1,N+1):
	if n%i==0:
		cnt+=1
	if cnt==K:
		print(i)
		break
else:
	print(-1)
  • for ~ else문은 for문을 예외없이 다 돌았을 때 ,즉 위의 문제에서는 조건을 만족하는 약수가 존재하지 않을 때 else:문이 실행

0개의 댓글