题目1534:数组中第K小的数字

很经典的一道算法题,我没想出来。。。

看了答案做出来的,二分法真强大,还有就是一定注意下边界条件,如果cal返回k则不一定是前k小,有可能没有这个数字,所以还要进一布减小。

#include <iostream>

#include <queue>
#include <iomanip>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <set>
#include <sstream>
#include <cstring>
using namespace std;
long long nums1[100001];
long long nums2[100001];
int m,n;
long long cal(long long  val)
{
    int j=n;
    long long add=0;
    for(int i=1;i<=m;i++)
    {
        if(j==0)
            break;
        while(j>=1&&(nums1[i]+nums2[j])>val)
            j--;
        add+=j;
    }
    return add;
}
int main()
{
    long long k;
    while(cin>>m>>n>>k)
    {
        for(int i=1;i<=m;i++)
            cin>>nums1[i];
        for(int i=1;i<=n;i++)
            cin>>nums2[i];
        long long mmin=nums1[1]+nums2[1];
        long long mmax=nums1[m]+nums2[n];
        int mid;
        while(mmin<=mmax)
        {
            mid=(mmin+mmax)/2;
            if(cal(mid)<k)
            {
                mmin=mid+1;
            }
            else if(cal(mid)>=k)
            {
                mmax=mid-1;
            }
        }
        cout<<mmin<<endl;
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值