若有N個無序數字,設計算法查找其中第K大的數字,並估計算法的複雜度

可以用簡單選擇排序或者冒泡找出第k大的元素,當N很大時時間複雜度趨近於O(N)
也可以用堆排序選出來,時間複雜度O(K Log N)
最優解使用快排,時間複雜度O(N)
用快排劃分成兩個子集,且當前界限爲第x大,若 x > k ,則要找元素爲x左邊子集中第x-k大的元素,若 x < k ,則要找元素在x右邊子集,爲第k大元素,若 x = k, 劃分元素即爲所求,代碼:

#include<stdio.h>
#include<stdlib.h>
int Partition(int A[],int low,int high){
	int pivot = A[low];
	while(low < high){
		while(low < high && A[high] > pivot) high--;
		A[low] = A[high];
		while(low < high && A[low] < pivot) low++;
		A[high] = A[low];
	}
	A[low] = pivot;
	return low;
}
int Select_k(int A[],int low,int high,int k){
	int index;
	if(low == high)	
		return A[low];
	int pivot = Partition(A,low,high);
	index = high - pivot + 1;//index標識當前劃分界限這個數第幾大
	if(index == k)
		return A[pivot];
	else if(index < k)
		Select_k(A,low,pivot-1,k-index);//從左邊找第k-index大的數 
	else
		Select_k(A,pivot+1,high,k); //從右邊找第k大的數 
}
int main(){
	int A[] = {10,9,8,7,6,5,4,3,2,1};
	printf("%d",Select_k(A,0,9,3));
	return 0;
} 
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章