選擇問題(分治策略)


選擇問題(Selection Problem),即在n個元素的集合中尋找第K小的元素的問題。第K小的元素又叫第K個順序統計量。有以下幾種變體: 
- 找最大值和最小值;同時找最大和最小值 
- 找中位數(第n/2小) 
- 找任意第K小的元素 
- 找Top-K的元素

 1 //選擇問題
 2 #include<iostream>
 3 #include<cstdio>
 4 #include<cstdlib>
 5 #include<algorithm>
 6 using namespace std;
 7 int a[1001];
 8 
 9 int select(int left,int right,int k){
10     if( left >= right )
11         return a[left];
12     int x = a[left];
13     int i = left;
14     int j = right+1;
15     while( true ){
16         do{
17             i++;
18         }while(a[i]<x);
19         do{
20             j--;
21         }while(a[j]>x);
22         if( i>=j )
23             break ;
24         swap(a[i],a[j]);
25     }
26     if( j-left+1 == k )
27         return x;
28     a[left] = a[j];
29     a[j] = x;
30     if( j-left+1 < k )
31         return select(j+1,right,k-j+left-1);
32     else
33         return select(left,j-1,k); 
34 }
35 
36 int main(){
37     int n;
38     while( scanf("%d",&n) != EOF ){
39         int k;
40         cin>>k;
41         for( int i = 0; i < n; i++ )
42             cin>>a[i];
43         cout<<select(0,n-1,k)<<endl;
44     }
45     return 0;
46 }


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM