堆排序练习：POJ 2388

关于堆排序请参看：http://128kj.iteye.com/blog/1679094
POJ2388题意：
【输入】第一行为n，接下来n行分别为一个数；
【输出】这n个数排序后的中位数
样例：
Sample Input

5
2
4
1
3
5
Sample Output

3

分析：好象用多种排序法都可以AC，这里先用堆排序，主要是复习一下堆排序代码。

排一次序后输出中位数，但效率太低了。这里先不管了。

Java代码

1. import java.util.Scanner;
2. public class Main {
4. public static void main(String[] args) {
5. Scanner in=new Scanner(System.in);
6. int n=in.nextInt();
7. int[] array =new int[n+1];
8. // array[0]=100; 不参与,下标从1开始
9. for(int i=1;i<=n;i++)
10. array[i]=in.nextInt();
11. heapSort(array);//堆排序
12. System.out.println(array[n / 2+1 ]);
13. //堆排序的结果
14. // for(int i=1;i<=n;i++)
15. // System.out.print(array[i]+" ");
16. }
18. //把a,b位置的值互换
19. public static void swap(int[] array, int a, int b) {
20. //临时存储child位置的值
21. int temp = array[a];
22. array[a]=array[b];
23. array[b]=temp;
24. }
26. /*将数组调整成堆
27. *根据树的性质建堆，树节点前一半一定是分支节点，即有孩子的，所以我们从这里开始调整出初始堆
28. */
29. public static void adjust(int[] array){
30. for (int i = array.length /2; i > 0; i--)
31. adjust(array,i, array.length-1);
34. }
36. /**
37. * 调整堆，使其满足堆得定义
38. * @param i
39. * @param n
40. */
41. public static void adjust(int[] array,int i,int n) {
43. int child;
44. for (; i <= n / 2; ) {
45. child = i * 2;
46. if(child+1<=n&&array[child]<array[child+1])
47. child+=1;/*使child指向值较大的孩子*/
48. if(array[i]< array[child]){
49. swap(array,i, child);
50. /*交换后，以child为根的子树不一定满足堆定义，所以从child处开始调整*/
51. i = child;
53. } else break;
54. }
55. }
57. //对一个最大堆heap排序
58. public static void heapSort(int[] array) {
59. adjust(array);//建堆
60. for (int i = array.length-1; i >0; i--) {
61. /*把根节点跟最后一个元素交换位置，调整剩下的n-1个节点，即可排好序*/
62. swap(array,1, i);
63. adjust(array,1, i - 1);
64. }
65. }
68. }