快速排序的一个特点是:每一次分区(partition)操作之后,就有一个元素被放在了数组的最终位置,在以后的排序过程中该元素位置不会变动;
利用这个特点我们可以将快速排序稍加改造来寻找第k个最小值,假设在一次分区操作之后中枢(pivot)的位置在k之前,那么我们下次只需要在中枢的后面进行查找;如果中枢的位置在k之后,那么我们下次只需要在中枢之前进行查找,直到中枢等于k为止。
我们知道快速排序的时间复杂度为O(nlgn),因为在寻找第k个最小的值时,我们只需要在中枢的一侧进行查找,所以通常来说这个算法的时间复杂度小于O(nlgn),但是,在最坏情况下这个算法的时间复杂度仍为O(n^2),比如在原数组有序时如{1,2,3,4,5,6,7,8,9,10...n},我们在查找第k个最小值的时候,每次中枢的位置都在第一个,而中枢最小,每次只排除一个元素,假设我们要查找第n/2个最小元素,总的比较次数为:(n-1)+(n-2)+...+(n/2-1),所以复杂度仍为O(n^2)。
代码实现如下,只需要在原快速排序上做少量改动,想了解快速排序可以看连接快速排序整理分析。
#include<stdio.h> #include<stdlib.h> int partition(int *a,int left,int right){ int i = left,j=right; int pivot = a[left]; while(i<j){ for(;i<j && a[j]>=pivot;j--); a[i] = a[j]; for(;i<j && a[i]<=pivot;i++); a[j] = a[i]; } a[i] = pivot; return i; } //寻找第k个最小值,k=1,2,...,n int quick_sort(int *a,int left,int right,int k){ int split_point; int value; if(left==right){ //当left==right时,返回a[left],这个原快速排序不需要 return a[left]; } if(left<right){ split_point = partition(a,left,right); if(split_point==k-1){ //找到了,返回值 return a[split_point]; } if(split_point>k-1){ //在左侧查找 value = quick_sort(a,left,split_point-1,k); }else{<span style="white-space:pre"> </span>//在右侧查找 value = quick_sort(a,split_point+1,right,k); } } return value; } void main(){ int a[] = {11,21,31,14,51,161,7,8,9,10}; int value = quick_sort(a,0,9,5); printf("%d",value); printf("\n"); }
时间: 2024-10-23 23:16:39