[백준_1300] K번째 수

문제 보기: https://www.acmicpc.net/problem/1300

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

 

풀이 과정

1. 이진 탐색을 이용한 문제 이전 문제도 이진 탐색이어서 같은 아이디어로 생각하였습니다.

2. int 형태로 변수 두 개를 입력받은 후 이 것을 이용하여 중간 값과 비교하여 탐색을 합니다.

3. n값은 이차원 배열과 1차원 배열의 크기를 나타내 줄 수 있는 값이기 때문에 end값을 구하는 데 사용하도록 합니다.

4. mid+1 값을 start값으로 설정하면서 반환 시켜서 원하는 값을 얻어줍니다.

 

코드

n = int(input())
k = int(input())

def binary_search(target, start, end):
    while(start <= end):
        mid = (start + end) // 2

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

        if cnt >= target:
            end = mid-1
        else:
            start = mid+1
    return start


print(binary_search(k,1,n*n))