5.查找最小的k個元素

 題目:輸入n個整數,輸出其中最小的k個。
例如輸入1,2,3,4,5,6,7和8這8個數字,則最小的4個數字爲1,2,3和4。
分析:這道題最簡單的思路莫過於把輸入的n個整數排序,這樣排在最前面的k個數就是最小的k個數。只是這種思路的時間複雜度爲O(nlogn)。我們試着尋找更快的解決思路。
我們可以開闢一個長度爲k的數組。每次從輸入的n個整數中讀入一個數。如果數組中已經插入的元素少於k個,則將讀入的整數直接放到數組中。否則長度爲k的數組已經滿了,不能再往數組裏插入元素,只能替換了。如果讀入的這個整數比數組中已有k個整數的最大值要小,則用讀入的這個整數替換這個最大值;如果讀入的整數比數組中已有k個整數的最大值還要大,則讀入的這個整數不可能是最小的k個整數之一,拋棄這個整數。這種思路相當於只要排序k個整數,因此時間複雜可以降到O(n+nlogk)。通常情況下k要遠小於n,所以這種辦法要優於前面的思路。

 


#include<iostream>
using namespace std;

void quicksort(int *a,int left,int right)
{
 int i=left;
 int j=right;
 int p=a[i];
 int temp;
 while(i<j)
 {
  while(i<j&&a[j]<=p)
   j--;
  if(i<j)
  {
   temp=a[j];
   a[j]=p;
   a[i]=temp;
   i++;
  }
  

  while(i<j&&a[i]>=p)
   i++;
  if(i<j)
  {
   temp=a[i];
   a[i]=p;
   a[j]=temp;
   j--;
  }
 }

 if(i!=left)
  quicksort(a,left,i-1);
 //if(j!=right)
  //quicksort(a,j+1,right);
}

void output(int *a,int length)
{
 for(int i=0;i<length;i++)
  cout<<a[i]<<" ";
}


void Mink(int *a,int length,int k)
{
 int *ret=new int[k];
 for(int i=0;i<k;i++)
  ret[i]=a[i];
 //quicksort(ret,0,k-1);
 for(int i=k;i<length;i++)
 {
  quicksort(ret,0,k-1);
  if(a[i]<ret[0])
   ret[0]=a[i];
 

 }
 output(ret,k);
 

}

int main()
{
 //int a[6]={9,4,2,7,5,1};
 //quicksort(a,0,5);
 //output(a,6);
 int a[10]={1,2,3,4,5,6,7,8,9,10};
 Mink(a,10,4);

}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章