백준 1300번

백준 1300번

K번째 수

시간 제한 : 2초 메모리 제한 : 128 MB

문제

세준이는 크기가 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]를 출력한다.

해설

먼저 k보다 작은 수의 곱이 몇개인지 찾으면 된다. k보다 작은 수가 몇개인지 찾아내면 k가 몇번째 숫자인지 알아낼 수 있기 때문이다.

코드의 mid 는 b 리스트를 일렬로 나열했을 때, cnt번째 값을 나타낸다. 즉, a 배열에서 mid 보다 작은 값의 개수가 cnt개인 것이다.

```python 3

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

left = 1 right = K ans = 0

while left <= right : cnt = 0 mid = (left+right) // 2

for i in range(1, N+1):
    cnt += min(mid // i, N)

if cnt < K :
    left = mid + 1
else :
    ans = mid
    right = mid - 1

print(ans)

```

Written on October 29, 2022