【算法设计与分析】线性时间内寻找元素（递归与分治）

最新推荐文章于 2022-11-23 20:01:05 发布

球王武磊

最新推荐文章于 2022-11-23 20:01:05 发布

阅读量1k

点赞数 5

分类专栏：算法设计与分析

本文链接：https://blog.csdn.net/weixin_46283740/article/details/115451575

版权

算法设计与分析专栏收录该内容

21 篇文章 19 订阅

订阅专栏

该博客介绍了如何使用递归与分治策略在线性时间内寻找数组中的第i小元素。通过分组和选取中位数作为基准，实现了快速的查找过程。样例展示了在一组整数中找到第3小元素的方法，并给出了相应的输出结果。代码实现部分包括了关键的Select和Partition函数，以及如何根据输入进行分组和查找。

摘要由CSDN通过智能技术生成

【算法设计与分析】线性时间内寻找元素（递归与分治）

【问题描述】
每次都是优化选出一个元素（分组后的中位数）为划分基准，在线性时间内寻找第i小元素。提示：分组时的组的个数为n/5的向下取整；分组后的中位数取第（num_group/2向上取整）小的元素。

【输入形式】
在屏幕上输入若干整数，各数间都以一个空格分隔。再输入要寻找的元素是数组从小到大顺序中第几个位置。

【输出形式】
输出第一次划分找到的基准元素以及数组从小到大顺序中要寻找的那个位置的元素。

【样例输入】

2 9 8 0 7 10 1 12 3 14 5 13 6 11 4
3

【样例输出】

7
2

【样例说明】
输入：15个整数，以空格分隔。要寻找第3小元素。
输出：
7，表示第一次划分得到的基准元素是7。
2，表示第3小元素为2。

【题解代码】

#include <iostream>
#include <cstdio>
#include <sstream>
#include <algorithm>
using namespace std;

int flag=1,a[1005];

void Swap(int &a,int &b)//位运算交换两个数 
{
	if(a==b) return;
	a=a^b;
	b=a^b;
	a=a^b;
}

int Partition(int a[],int p,int r,int x)//划分 
{
	int pos=p;
	for(int i=p;i<=r;i++)
		if(a[i]==x)
		{
			pos=i;
			break;
		}
	Swap(a[p],a[pos]);
	int i=p,j=r+1;
	while(true)
	{
		while(a[++i]<x&&i<r);
		while(a[--j]>x);
		if(i>=j) break;
		Swap(a[i],a[j]);
	}
	a[p]=a[j];
	a[j]=x;
	return j;
}

int Select(int a[],int p,int r,int k)//Select算法 
{
	int n=r-p+1;
	if(n<5)//待找范围元素少于5个 
	{
		sort(a+p,a+r+1);
		return a[p+k-1];
	}
	for(int i=0;i<=(n-5)/5;i++)//分组时的组的个数为n/5的向下取整 
	{
		sort(a+p+5*i,a+p+5*i+5);
		Swap(a[p+5*i+2],a[p+i]);
	}
	int x=Select(a,p,p+(n-5)/5,(n/5+1)/2);//找中位数的中位数,分组后的中位数取第（num_group/2向上取整）小的元素 
	if(flag==1)
	{
		cout<<x<<endl;
		flag=0;
	}//输出第一次划分找到的基准元素 
	int q=Partition(a,p,r,x);
	int j=q-p+1;
	if(k==j)
		return a[q];
	else if(k<j)
		return Select(a,p,q-1,k);
	else
		return Select(a,q+1,r,k-j);
}

int main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	stringstream ss;
	string s;
	getline(cin,s);
	ss<<s;
	int t,k,num=0;
	while(ss>>t)
		a[num++]=t;
	cin>>k;
	int ans=Select(a,0,num-1,k);
	cout<<ans<<endl;
	return 0;
}