(Python/🥇1)백준알고리즘 1300번: K번째 수

 

문제 바로가기 

 

문제:

세준이는 크기가 N×N인 배열 A를 만들었다. 배열에 들어있는 수 A[i][j] = i×j 이다. 이 수를 일차원 배열 B에 넣으면 B의 크기는 N×N이 된다. B를 오름차순 정렬했을 때, B[k]를 구해보자. 배열 A와 B의 인덱스는 1부터 시작한다.

입력:

첫째 줄에 배열의 크기 N이 주어진다. N은 105보다 작거나 같은 자연수이다. 둘째 줄에 k가 주어진다. k는 min(109, N2)보다 작거나 같은 자연수이다.

출력:

B[k]를 출력한다.

풀이:

import sys

sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')

N, K = int(input()), int(input())

start = 1; end = K
res = 0
while start <= end:
    # middle 값
    mid = (start + end) // 2
    temp = 0
    # 각 행마다 mid 숫자보다 작은 갯수 temp에 다 더함
    for i in range(1, N+1):
        temp += min(mid // i, N)
    # 갯수가 많으면 줄여줌
    if temp >= K:
        res = mid
        end = mid-1
    else:
        start = mid +1
print(res)

 

 

1,2,3,4,5

2,4,6,8,10

3,6,9,12,15

4,8,12,16,20

5,10,15,20,25

 

이렇게 있을때 각 행마다 mid 숫자보다 작은 갯수를 temp에 더해서 순번을 찾아내면 된다.