K번째 수

시간 제한메모리 제한제출정답맞힌 사람정답 비율

2 초

128 MB

47632

17678

12985

37.838%

문제

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

A배열은 N행이 N의 배수로 구성되어 있다.(A[i][j]=ixj)

k번째 수는 k를 넘지 않는다. 시작은 1, 종료는 k로 설정한다.

#include <iostream>

using namespace std;

int main()
{
	int N, K;
	cin >> N >> K;

	long start = 1;
	long end = K;
	long answer = 0;

	while (start <= end)
	{
		long mid = (start + end) / 2;
		long target = 0;

		for (int i = 1; i <= N; i++)
		{
			if (mid / i < N)
			{
				target += mid / i;
			}
			else
			{
				target += N;
			}
		}

		if (target < K)
		{
			start = mid + 1;
		}
		else
		{
			answer = mid;
			end = mid - 1;
		}
	}

	cout << answer;
}