算法導論--在無序數組中選擇第k小個數

從一個無序的整型數組中選出第k小的數，如k=1爲最小數，k=n爲最大數。這裏數組可以是有重複的值！

下面是自己寫的一個函數，記在此處來記憶我留下的痕跡！

 

//選擇無序數組中第k小的數
#include <iostream>
using namespace std ;
bool failed = false ;
//這裏只考慮數組是int型的
int findnumber(int *array,int start , int end, int k)
{
    if(array == NULL || start > end || k < start || k > end+1 || k <= 0 )
    {
        failed = true ;
        return 0;
    }
    if(start == end)
    {
        return array[start] ;
    }
    int len = end - start + 1 ;
    int tmp = 0 ;
    int ps = rand()%len +start ;
    int tk = k ;
   
    while(true)
    {
      //分割兩數組
      int f = start ;
      int t = array[ps] ;
      int equalnum = 0 ;
      for(int i = start ; i <= end ; i ++ )
      {                
               if(array[i]< t )
               {
                  
                   tmp = array[f];
                   array[f] = array[i];
                   array[i] = tmp ; 
                   f ++ ;                           
               }else if(array[i] == t)
               {
                   tmp = array[f];
                   array[f] = array[i];
                   array[i] = tmp ; 
                   f ++ ; 
                   equalnum ++ ;
               }
      
       } //end
       f--;
       if(equalnum > tk  && (f - start + 1) == equalnum)
       {
            return t ;//這裏是記錄數據相等的數目，當我們從開始start處到最後處end都被這個值給充斥了，那麼肯定是這裏面的值了，再進行下去就會陷入死循環了。
       }
       if(tk == (f - start + 1) ) 
       {
           return t ;
        }
        if((f - start + 1 ) > tk )
        {
            end = f ;
        }else
        {
             start = f + 1   ;
             tk = k - start   ; //這個地方犯過錯誤，就是寫成了k=k-start，在調試的時候老發現無限的循環。後來打印k的值的時候發現k的值都***爲負了。這個bug，這個過錯使得在一次運行可能會得到正確的數據，但是多次運行後程序就崩潰。
         }
         len = end - start + 1  ;
         ps = rand()%len +start ;
        
    }
}

int main()
{
    int array[10] = {1,1,1,2,2,1,4,1,1,1};
    for(int i = 0 ; i < 10 ; i ++ )
    {
        cout<<findnumber(array,0,9,i+1)<<endl; 
    }
    system("pause");
    return 0 ;
}

 

先想好，分析好問題，自己腦中構思好了編寫的思路，且想好了程序出錯的地方再編程，這樣會快的很多，而不是一看到問題就框框的在電腦上敲。