减治算法之寻找第K小元素问题

最新推荐文章于 2022-03-17 23:02:03 发布

tterminator

最新推荐文章于 2022-03-17 23:02:03 发布

阅读量3.2k

点赞数

CC 4.0 BY-SA版权

分类专栏： Algorithm Design AlgorithmDesign 文章标签：算法

本文链接：https://blog.youkuaiyun.com/tterminator/article/details/50966927

AlgorithmDesign 同时被 2 个专栏收录

12 篇文章

订阅专栏

Algorithm Design

11 篇文章

订阅专栏

本文介绍了一种基于快速排序思想的算法，用于在未排序的整数数组中找到第K小的元素。该算法首先通过分区操作部分排序数组，然后递归地搜索目标元素，直至找到位于第K位的数值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、问题描述

给定一个整数数列，寻找其按递增排序后的第k个位置上的元素。

二、问题分析

借助类似快排思想实现pation函数，再利用递归思想寻找k位置。

三、算法代码

public static int selectMinK(int [] arr, int low, int high, int k){
		int index = pation(arr, low, high);
		if(index == k){
			return arr[index];
		}
		if(index < k){
			return selectMinK(arr, index + 1, high, k);
		}else{
			return selectMinK(arr, low, index - 1, k);
		}
	}
	
	public static int pation(int [] arr, int low, int high){
		while(low < high){
			while(low < high && arr[low] <= arr[high]){//从后往前，把小的元素往前调换
				high--;
			}
			if(low < high){
				int tmp = arr[low];
				arr[low] = arr[high];
				arr[high] = tmp;
				low++;
			}
			while(low < high && arr[low] <= arr[high]){//从前往后，把大的元素往后调换
				low++;
			}
			if(low < high){
				int tmp = arr[low];
				arr[low] = arr[high];
				arr[high] = tmp;
				high--;
			}
		}
		return low;//返回low，high相遇位置
	}

四、完整测试代码

public class Solution {
	
	public static void main(String [] args){
		int [] randArr = new int[]{5,2,8,6,3,6,9,7};
		int result = selectMinK(randArr, 0, randArr.length - 1, 4);
		System.out.print(result);
	}
	public static int selectMinK(int [] arr, int low, int high, int k){
		int index = pation(arr, low, high);
		if(index == k){ //若返回的下标为k，则找到目标元素
			return arr[index];
		}
		if(index < k){
			return selectMinK(arr, index + 1, high, k);
		}else{
			return selectMinK(arr, low, index - 1, k);
		}
	}
	
	public static int pation(int [] arr, int low, int high){
		while(low < high){
			while(low < high && arr[low] <= arr[high]){
				high--;
			}
			if(low < high){
				int tmp = arr[low];
				arr[low] = arr[high];
				arr[high] = tmp;
				low++;
			}
			while(low < high && arr[low] <= arr[high]){
				low++;
			}
			if(low < high){
				int tmp = arr[low];
				arr[low] = arr[high];
				arr[high] = tmp;
				high--;
			}
		}
		return low;
	}
}

五、运行结果

第4小元素为：6