算法实验之分治法求中位数

利用分治策略试设计一个O (log n)时间的算法求出这2n个数的中位数。

要输入的内容在文件1.txt中，输出的结果在文件2.txt中。

[cpp] view plaincopy

1. #include<iostream>  
2. #include<fstream>  
3. using namespace std;  
4. template <class T>  
5. T mid(T *a,T *b,int len)  
6. {  
7.     if(len==1)  
8.         return *a<*b?*a:*b;  
9.     if(len==2)  
10.     {  
11.         int temp=*a>=*b?*a:*b;  
12.         if(temp==*a)  
13.             return temp<*(b+1)?temp:*(b+1);  
14.         else  
15.             return temp<*(a+1)?temp:*(a+1);  
16.     }  
17.     int n;  
18.     if(len%2==0)  
19.         n=len/2;  
20.     else  
21.         n=len/2+1;  
22.     if(*(a+n-1)==*(b+n-1))  
23.         return *(a+n-1);  
24.     else if(*(a+n-1)<*(b+n-1))  
25.         return mid(a+n-1,b,n);  
26.     else if(*(a+n-1)>*(b+n-1))  
27.         return mid(a,b+n-1,n);  
28.   
29. }  
30. int main()  
31. {  
32.     int a[5];  
33.     int b[5];  
34.     ifstream f1("1.txt");     
35.     ofstream f2("2.txt");  
36.     int len;  
37.     f1>>len;  
38.     for(int i=0;i<len;i++)  
39.         f1>>a[i];   
40.     for(i=0;i<len;i++)  
41.         f1>>b[i];  
42.     int *pa=a;  
43.     int *pb=b;  
44.     int m=mid(pa,pb,len);  
45.     cout<<m<<endl;  
46.     f2<<m;  
47.     f1.close();  
48.     f2.close();  
49.     return 1;  
50. }  

运行结果：