#P1923. 【深基9.例4】求第 k 小的数

#include <iostream>
#include <algorithm>
using namespace std;
int a[5000001];
    int n,k;
void dive(int l,int r)
{
    int i=l,j=r,mid=a[(l+r)/2];
    while(i<=j)
    {
        while(a[i]<mid) i++;//一直i++直到遇到第一个大于等于mid的数a[i] 
        while(a[j]>mid) j--;//一直j--直到遇到第一个小于等于mid的数a[j] 
        if(i<=j)
        {
            swap(a[i],a[j]);//交换在mid前遇到的比mid大的数和在mid后遇到的比mid小的数
            i++;
            j--;
        }
    }// 使得mid前面的数都小于mid,mid后面的数都大于mid(mid的值在循环中也可能发生变化 
    //对于k的值分三种情况 
    if(k<=j)//k在j前面 
    dive(l,j);
    else if(k>=i)//k在i后面 
    dive(i,r);
    else if(k>=j&&k<=i)//k在i,j中间(k=i=j),a[k]=mid 
    return;
}
int main()
{        
        std::ios::sync_with_stdio(false);
    cin >> n >> k;
    for(int i=0;i<n;i++)
    {
        cin >> a[i];
    }
    dive(0,n-1);

    cout <<a[k] << endl;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值