CF448D - Multiplication Table
题目
Multiplication Table - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
思路
-
题目要求的是矩阵中第k小的数,即从小到大排序后第k个数。因为n * m的数量级在10的11次方,显然不能进行排序求解。因为f(i,j)= i * j ,矩阵中的每一行都是有序的,设答案为ans我们可以求出每一行小于或等于ans数的个数,这一行的数为i * (1 ~ n),所以小于或等于ans的数为m / i个,因为m / i可能大于m所以count = min(m, m / i)。
-
我们可以用枚举1 ~ n计算所有行小于或等于ans的数的和记为sum,当sum >= k时说明答案等于或小于当前数,反之大于当前数,所以可以利用这个性质对答案进行二分,l = 1, r = n * m,结束条件l > r。
-
当sum >= k,r = mid - 1,当sum < k,l = mid + 1。
-
我们要注意的是答案可能不在矩阵中,举例当n = 3, m = 3, k = 11时,
矩阵为:
1 2 3 4
2 4 6 8
3 6 9 12 -
可以看到当ans = 9时满足条件,当我们二分时可能会得到ans = 10或11(仅为了说明这种情况)也满足,即ans = 9,10,11得到的sum是一样的,而10,11这两个答案显然不合法,而我们要解决这个问题就要保证在同样sum的条件下取最小的ans。我们只需要在sum < k 的时候更新l = mid + 1,当二分结束时得到l即为最小的答案。
代码
// 因为n * m的数量级在10的11次方,所以变量最好用long long
#include <bits/stdc++.h>
using namespace std;
// 定义long long为LL
typedef long long LL;
// 定义所需变量
LL n, m, k;
// 判段小于等于x数的个数sum是否大于等于k
bool check(LL x) {
// 初始化sum为0
LL sum = 0;
// 枚举从1 ~ n行
for (int i = 1; i <= n; i++) {
// 累加每一行小于x的个数
sum += min(m, x / i);
}
// 返回sum >= k是否成立
return sum >= k;
}
int main() {
// 读入数据
cin >> n >> m >> k;
// 初始化二分区间,l = 1,r = n * m
LL l = 1, r = n * m;
// 当l > r时跳出循环
while (l <= r) {
// 计算区间中位数,利用减法防止爆
LL mid = l + (r - l) / 2;
// 当sum >= k
if (check(mid)) {
// 考虑剩下的[l, mid - 1]的是否有更小的答案
r = mid - 1;
} else {
// 当sum < k,说明答案一定比l大
l = mid + 1;
}
}
// 输出答案
cout << l << endl;
return 0;
}
复杂度分析
- 时间复杂度O(n * log(n*m))
- 空间复杂度O(1)