[Python] 백준 1300 K번째 수

백준 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]를 출력한다.

---

제출

n,k=int(input()),int(input())
s,e=1,k
while s<=e:
    m=(s+e)//2
    t=0
    for i in range(1,n+1):
        t+= min(m//i,n)
    if t>=k:
        a=m
        e=m-1
    else:
        s=m+1
print(a)

예제

3
7

 

결과

백준 1300 K번째 수

 

1300번: K번째 수