QA & Engineering Blog

A Blog about Quality · Automation · Engineering

🏠 홈으로

[Gold II] K번째 수 - 1300

문제 링크

성능 요약

메모리: 2020 KB, 시간: 12 ms

분류

이분 탐색, 매개 변수 탐색

문제 설명

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

💡 Solutions

📄 K번째 수.cc

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>

#include <stdio.h>

#define _CRT_SECURE_NO_WARNINGS
// 즉 위의 중복되는 변수를 읽어오는 것에 문제가 있을 수 있다,.
using namespace std; // - 이걸 활용해서 네임스페이스를 자동으로 선언함// but 원래의 목적을 상실
// using std::count; 
// using std::endl; // 이렇게 from - import 느낌으로도 사용 가능


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

	vector<int> B;

	int start = 1;
	int end = K;
	int result;
	while (start <= end)
	{
		int sum = 0;
		int mid = (start + end) / 2; // 숫자 Num 은 K 보다 작거나 같다.
		for (int i = 1; i <= N; i++)
		{
			sum += min(N, mid / i);
		}

		if (sum < K) {
			start = mid + 1;
		}
		if (sum >= K) {
			end = mid - 1;
			result = mid;
		}
	}

	cout << result;
	return 0;
}