IT-garden

문제

세준이는 크기가 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=int(input())
k=int(input())
 
low=1
high=k

while low <= high:
    mid = (low + high)//2
    count = 0
    for i in range (1, n+1):
        count = count + min(mid//i, n)
 
    if count < k:
        low = mid + 1
    else:
        high = mid - 1
print(low)

백준 알고리즘 1300번 : https://www.acmicpc.net/problem/1300

문제

수 N개 A1, A2, ..., AN이 주어진다. A를 오름차순 정렬했을 때, 앞에서부터 K번째 있는 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N(1 ≤ N ≤ 5,000,000)과 K (1 ≤ K ≤ N)이 주어진다.

둘째에는 A1, A2, ..., AN이 주어진다. (-109 ≤ Ai ≤ 109)

출력

A를 정렬했을 때, 앞에서부터 K번째 있는 수를 출력한다.

예제 입력과 출력

알고리즘 분류

정렬

정답

import sys

n,k=sys.stdin.readline().split()
a=list(map(int,sys.stdin.readline().split()))

a.sort()

print(a[int(k)-1])

백준 알고리즘 11004번 : www.acmicpc.net/problem/11004