归并排序

 1  /* *
 2   * 归并排序
 3   * yaoC 2014-09-03
 4   * Θ(nlg(n))
 5    */
 6 
 7 #include<iostream>
 8  using  namespace std;
 9 
10  #define infinity 999999 // 定义无穷大
11 
12  /* *
13   * [merge 合并]
14   * @param A [待合并数组，也是合并后的数组]
15   * @param p [数组第一部分开始下标]
16   * @param q [数组第一部分结束下标]
17   * @param r [数组第二部分结束下标]
18    */
19  void merge( int* A, int p, int q, int r)
20 {
21      // 分为两个数组，增加最后一位值为无穷大
22       int n1=q-p+ 1;
23      int n2=r-q;
24      int* A1= new  int[n1+ 1];
25      int* A2= new  int[n2+ 1];
26      for ( int i =  0; i < n1; i++)
27         A1[i]=A[p+i];
28     A1[n1]=infinity;
29      for ( int i =  0; i < n2; i++)
30         A2[i]=A[q+ 1+i];
31     A2[n2]=infinity;
32      // 合并
33       int i= 0,j= 0,k=p;
34      while(k<=r)
35     {
36          if(A1[i]<A2[j])
37         {
38             A[k]=A1[i];
39             i++;
40         }
41          else
42         {
43             A[k]=A2[j]; 
44             j++;
45         }
46         k++;
47     }
48 }
49  /* *
50   * [mergeSort 分解]
51   * @param A [待排序数组]
52   * @param p [数组开始下标]
53   * @param r [数组结束下标]
54    */
55  void mergeSort( int* A, int p, int r)
56 {    
57      int q=(p+r)/ 2;
58      // 将数组一分为二
59       if(p<q)
60     {
61         mergeSort(A,p,q);
62         mergeSort(A,q+ 1,r);
63     }
64     merge(A,p,q,r); // 合并两个数组
65  }
66  // test
67  int main()
68 {
69      int num =  0;  
70     cout <<  " 个数： ";
71     cin >> num;
72      int* p =  new  int[num];
73     cout <<  " 数值： ";
74      for ( int i =  0; i < num; i++)
75         cin >> p[i];
76     mergeSort(p, 0,num- 1);
77     cout <<  " after sorting: ";
78      for ( int i =  0; i < num; i++)
79         cout << p[i] <<  "   ";
80     cout << endl;
81 }