二维表第k大数-拼多多2020校招

最新推荐文章于 2024-04-29 17:03:19 发布

sigd

最新推荐文章于 2024-04-29 17:03:19 发布

阅读量119

点赞数

分类专栏：分治算法文章标签： c++ 算法

本文链接：https://blog.csdn.net/sigd/article/details/130308216

版权

分治算法专栏收录该内容

5 篇文章 0 订阅

订阅专栏

该程序通过二分查找算法解决在n*m的二维数组中找到第k小的数的问题。由于可能存在相同的元素，至少k个用于考虑重复值的情况。转换目标为找第k大，即第(n*m-k+1)小，然后在区间[1,n*m]中进行二分查找，检查mid值左侧是否至少有k个数小于等于mid。

摘要由CSDN通过智能技术生成

分析：第k大不好处理，第k小有一定规律。第k大就是第n*m-k+1小。二分法找寻满足条件的最小值。mid条件就是至少有k个数小于等于mid。为什么是至少有k个（而不是正好k个？），因为二维表中有相同元素，例如

1 2 3 4

2 4 6 8

3 6 9 12

4 8 12 16

其中12既是第14小，也是第15小。如果求第14小时，我们计算得到小于等于12的数字有15个，而小于等于11的数字只有13个，所以12能满足条件。注意此题目必须开longlong。

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
ll i,j,k,m,n;
int check(ll mid)/**< 计算mid 是第几小 */
{
    int s=0;/**<统计每一行有多少个比mid小的数字*/
    for(i=1;i<=n;i++)/**< 每一行最大值为i*m */
       s+=min(m,mid/i); /**< m表示这一行所有数字都比mid小 */
    return s>=k;
}
int main()
{
    cin>>n>>m>>k;
    k=m*n-k+1;/**< 转换成求第k小 */
    ll l=1,r=m*n,mid,best;
    while(l<=r)
    {
        mid=(l+r)/2;
        if(check(mid))
            best=mid,r=mid-1;
        else
            l=mid+1;
    }
    cout<<best;
    return 0;
}