[백준] 1300번 : K번째 수 (C++)

10984번: 내 학점을 구해줘

---

문제)

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

 

 

 

 

 

풀이)

 

#include <iostream>
#include <algorithm>
 
using namespace std;
 
long long n, k;
 
bool check(long long x)
{
    long long total = 0;
    for (int i = 1; i < n + 1; i++)
    {
        total += min(n, x / i);
    }
 
    return total >= k;
}
 
int main()
{
    cin >> n >> k;
 
    long long s = 1;
    long long e = n * n;
    
    long long ans;
    while (s <= e)
    {
        long long mid = (s + e) / 2;
 
        if (check(mid))
        {
            ans = mid;
            e = mid - 1;
        }
        else
        {
            s = mid + 1;
        }
    }
    
    cout << ans;
 
    return 0;
}
 

---

출처 : https://www.acmicpc.net/problem/1300 

1300번: K번째 수