想一想快速排序的思想:将数组中某一个元素m作为划分依据(我们假设为第一个元素,即m = arr[0]),遍历一遍数组,使得数组的格局变成这样的三个部分:(1)m前面的元素 (2)m (3)m后面的元素。其中m前面的元素小于m,m后面的元素大于m,这样找第k小的数正好可以借鉴这个思想,即:
1、若m前面的元素个数大于k,则第k小的数一定在m前面的元素中,这时我们只需要继续在m前面的元素中搜索第k小的数;
2、若m前面的元素个数小于k,则第k小的数一定在m后面的元素中,这是我们只需要继续在m后面的元素中搜索第k-s小的数,其中s是m前面的元素个数。
#include <iostream.h> #define MAX_SIZE 100 int Biger[MAX_SIZE]; int Smaller[MAX_SIZE]; int Select_kth_Small(int arr[],int n,int k) { if(n == 1) return arr[0]; int b = 0,s = 0,t = arr[0],temp_n,temp_k; int temp[MAX_SIZE]; for(int i = 1 ; i < n ; i++)//遍历集合 { if(arr[i] > t) Biger[b++] = arr[i]; //如果当前元素比t大,就将当前元素加入Biger[] else Smaller[s++] = arr[i];//反之就加入到Smaller,这里没有考虑set[0] } if(b == 0) { Biger[b++] = t;//if...else主要是为了防止t大于或小于其他所有元素的情况 } else { Smaller[s++] = t; } //如果Smaller集合中的元素个数大于K,说明第K小的元素必在其中 //否则一定在Biger中,且应该是Biger集合中第k-r小的元素 //更新相应的变量 if(s >= k) { temp_n = s; temp_k = k; for(i=0;i<temp_n;i++) { temp[i] = Smaller[i]; } } else { temp_n = b; temp_k = k-s; for(i=0;i<temp_n;i++) { temp[i]=Biger[i]; } } return Select_kth_Small(temp,temp_n,temp_k); } int main() { int arr[]={4,0,1,0,2,3}; int ans = Select_kth_Small(arr,6,3); cout<<"在数组arr[]={4,0,1,0,2,3}中,第3小的数是:"<<ans<<endl; return 0; }
一定要学好数据结构和算法。