选第k小元素

最新推荐文章于 2023-05-03 09:58:36 发布

hereiszzw

最新推荐文章于 2023-05-03 09:58:36 发布

阅读量131

点赞数 1

文章标签：算法

本文链接：https://blog.csdn.net/vvvii_/article/details/116010927

版权

选第k小元素：特定分治策略

1.问题

在一个无序数组中寻找第k小的元素

2.解析

选择select:
选取一个元素m作为主元
分区partiton:
根据m划分序列，使得m左边的元素都比它小，右边的元素都比它大.左边部分序列记为S1，右边部分序列记为S2
分治divide：
根据给定的第k小，若m左边的元素数量大于k，那么可在S1部分寻找第k小的元素，若左边的元素数量小于k，那么可在S2寻找第k小的元素。若k=S1+1,则m就是所要寻找的第k小的数。
递归recursion：
若k！=S1+1，则重复divide操作，不断递归下去，直到找到第k小的元素。

3.设计

#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 1000000;
int t,m,k;
int Find(int a[],int left,int right,int k) {
	if (left==right) return a[left];
	int i = left, j = right;
	int m = a[i];
	while (i < j) {
		while (i<j && a[j]>=m){
			j--;
		}
		a[i] = a[j];
		while (i<j && a[i]<=m){
			i++;
		}
		a[j] = a[i];
	}
	a[i] = m;
	int count = i-left+1;
	if (count < k) 
	    return Find(a,i+1,right,k-count);
	return Find(a,left,i,k);
}

int main() {
	cout<<"请输入元素个数t和k的值；"<<endl; 
	cin>>t>>k;
	int a[10000];
	for (int i=1;i<=t;i++) {
		cin>>a[i];
	}
	cout<<Find(a,1,t,k)<<endl;
	return 0;
}
/*
5
3
5 6 2 4 1
*/