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

题目描述

输入 nn（1 \le n < 50000001≤n<5000000 且 nn 为奇数）个数字 a_iai​（1 \le a_i < {10}^91≤ai​<109），输出这些数字的第 kk 小的数。最小的数是第 00 小。

请尽量不要使用 nth_element 来写本题，因为本题的重点在于练习分治算法。

输入格式

无

输出格式

无

输入输出样例

输入 #1复制

5 1
4 3 2 1 5

输出 #1复制

2

解题过程：

第一遍用了快排的这个模板：

#include<bits/stdc++.h>
using namespace std;
const int N=5e6+10 ; 
int k,n,a[N];
void qsort(int l,int r)
{
    if(l>=r) return;
    int i=l-1,j=r+1,x=a[l+r>>1];
    while(i<j)
    {
        do j--; while(a[j]>x);
        do i++; while(a[i]<x);        
        if(i<=j) swap(a[i],a[j]); 
    }
    if(k<j) qsort(l,j);
    else if(k>j) qsort(j+1,r);
    else
    {
        printf("%d\n",a[k]);
        return;
    }
}
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++) scanf("%d",&a[i]);
    qsort(0,n-1);
    return 0;
}

总是有一个点过不了，后面仔细思考以后，这个快排的思想只能保证 l<=j<=i<=r,其中（l,j）<=x和(j+1,r)>=x,j这个点的数据并不是(l,j)这个区间的最大的，因此直接输出j点有问题。需要使用快速排序的思想，但要归位的写法。

#include<bits/stdc++.h>
using namespace std;
const int N=5e6+10 ; 
int k,n,a[N];
void qsort(int l,int r)
{
	if(l>r) return;
	int i=l,j=r,x=a[l];
	while(i<j)
	{
		while(i<j&&a[j]>=x) j--; a[i]=a[j];
		while(i<j&&a[i]<=x) i++; a[j]=a[i];	
		//if(i<j) swap(a[i],a[j]); 
	}
	a[i]=x;	
	if(k<i) qsort(l,i-1);
	else if(k>i) qsort(i+1,r);
	else
	{
		printf("%d\n",a[i]);
		return;
	}
}
int main()
{
	//freopen("text.txt","r",stdin); 
	scanf("%d%d",&n,&k);
	for(int i=0;i<n;i++) scanf("%d",&a[i]);
	qsort(0,n-1);
	//for(int i=0;i<n;i++) printf("%d ",a[i]);
	return 0;
}