最小K元素之分治法——输入有坑！！！！

题目

输入一个整数数组，请求出该数组的第k小元素。要求时间复杂度为O(n)。

思路：

这道题思想是用分治法通过递归实现，借用了快排的原理，通过快排中分区函数得到的基准元素，得到基准元素前面的元素比基准元素小，后面的比基准元素大，这样通过统计基准元素前面的个数判断目标第K小元素是在基准元素前面还是后面，然后根据这进行递归，直到找到第K元素。但是这题在输入数据中设置了小小的急转弯，需要设计输入的方式

递归函数

int solve(int low,int high,int k)
{
    int s=part(low,high);//分区函数，得到基准元素的下标
    if(low==high)//递归结束的条件，当只剩下一个数时，也就是K元素
        return a[low];//返回结果
    int j=s-low+1;//得到基准元素前面元素的个数
    if(k<=j)//当k小于等于基准元素前面的个数，便在前面找
        return solve(low,s,k);
    else//当k大于基准元素前面的个数，便在后面找
        return solve(s+1,high,k-j);
}

分区函数

这里不再多说，可以去快排去理解

int part(int low,int high)
{
    int p=Random(low,high);//随机化选取基准函数
    Swap(p,low);//交换函数，交换a[p]，a[low]的值
    int j=low;
    int k=a[low];
    for(int i=low+1;i<=high;i++){
        if(a[i]<k){
            j++;
            Swap(j,i);
        }
    }
    Swap(j,low);
    return j;
}

全部代码

因为这道题输入时有困难，需要进行设计，而且是多组输入数据，需要发费些时间进行设计，可以用java一行输入字符串然后转数字（因为java没学好，没有实现），也可以用一个数组和2个字符进行输入，字符用来判断是否换行输入了一行，输入了一行后接受数据K，然后换行，此时第二个字符接受换行符，以便进行下组测试进行，此时不要忘记i清零

#include<bits/stdc++.h>
#define ll long long
#define Random(a,b) (rand()%(b-a+1)+a)
#define Swap(i,j) {int k;k=a[i];a[i]=a[j];a[j]=k;}
using namespace std;
const int N=100000;
int a[N];
int titl;


int part(int low,int high)
{
    int p=Random(low,high);
    Swap(p,low);
    int j=low;
    int k=a[low];
    for(int i=low+1;i<=high;i++){
        if(a[i]<k){
            j++;
            Swap(j,i);
        }
    }
    Swap(j,low);
    return j;
}
int solve(int low,int high,int k)
{
    int s=part(low,high);
    if(low==high)
        return a[low];
    int j=s-low+1;
    if(k<=j)
        return solve(low,s,k);
    else
        return solve(s+1,high,k-j);
}

int main(){

	int k,num,i=0;
	//cout<<1e6<<endl;
	char c,c2;
	while(~scanf("%d%c",&a[++i],&c)){//每次一个数字一个空格输入
        if(c=='\n'){//如果是换行符则第一行数据输入完毕，开始接收第二行数据
            cin>>num;//接受第二行K
            c2=getchar();//接收换行符
            cout<<solve(1,i,num)<<endl;//输出结果
            i=0;//i清零以便进行下组测试

        }
	}



	return 0;
}