第k个数
题意
给定一个 \(n \times m\) 的方格矩阵,第 \(i\) 行 \(j\) 列的方格内元素为 \(i \times j\) (行和列都从 \(1\) 开始编号)。
问这 $n \times m $ 个数中第 \(k\) 小的数字为多少。
\(1 \le n, m \le 5 \times 10^5, 1 \le k \le n \times m\) 。
分析
对于第 \(k\) 小的数字,整个矩阵中至少有 \(k\) 个数字小于等于它。
二分枚举答案 \(x\) ,对于第 \(i\) 行,一共有 \(min(m, x / i)\) 个数字小于等于它。
Code
#include <iostream>
using namespace std;
typedef long long ll;
ll n, m, k;
bool check (ll x)
{
ll res = 0;
for (int i = 1; i <= n; i ++ )
res += min(m, x / i);
return res >= k;
}
int main ()
{
cin >> n >> m >> k;
ll l = 1, r = n * m;
while(l < r)
{
ll mid = l + r >> 1;
if (check(mid)) r = mid;
else l = mid + 1;
}
cout << r << endl;
return 0;
}