首页 > 代码库 > 堆排序

堆排序

堆的定义如下:

n个元素的序列{k0,k1,...,ki,…,k(n-1)}当且仅当满足下关系时,称之为堆。

"ki<=k(2i),ki<=k(2i+1);或ki>=k(2i),ki>=k(2i+1).(i=1,2,…,[n/2])"

若将和此次序列对应的一维数组(即以一维数组作此序列的存储结构)看成是一个完全二叉树,

若完全二叉树中每一个节点的值都大于或等于任意一个子节点的值(如果有的话),称之为大顶堆。

若完全二叉树中每一个节点的值都小于或等于任意一个子节点的值(如果有的话),称之为小顶堆。

由此,若序列{k0,k1,…,k(n-1)}是堆,则堆顶元素(或完全二叉树的根)必为序列中n个元素的最小值(或最大值)。

倘若给堆中每一个节点都赋予一个整数值标签,根节点被标记为0,对于每一个标记为i的节点,其左子节点(若存在的话)被标记为2*i+1,其右子节点(若存在的话)被标记为2*i+2,对于一个标记为i的非根节点,其父节点被标记为(i-1)/2。使用这个标记,我们能够将堆存储在数组中,节点存储在数据中的位置就使其标签。

堆排序示例:

规则:

1.初始化i=0,先后查看节点i的左子节点2*i+1、右子节点2*i+2,如果发现违反了堆的定义,则交换2个节点的值(编号只跟位置有关,也就是数组中的下标),置i=0

2.如果没有发现违反堆的定义,则置i=i+1,返回步骤1

3.当i==n的时候,将0号位置上的元素放置到n号位置上,置i=0,置n=n-1,返回步骤1,对位置为[0, n]范围内的元素重复上述操作

示意图如下:数组中的值依次是17,8,45,84,2,94,这次排序的最终结果是把最大的94放到了序列末尾

下面是剩下的元素的排序过程:

代码如下:

 1 package sorts; 2  3 public class  Heap // 小顶堆,也就是说小的数据在堆顶部 4 { 5      6      7     public void heap_sort(int[] a,int n){ 8         if(n>0){ 9             init_sort(a,n);//初始化堆,找出最大的放在堆顶10             swap(a, 0, n);11             heap_sort(a, n-1); // recursively12         }else{13             print(a);14         }15     }16     17     private void swap(int[] a, int index1, int index2) {18         a[index1] = a[index1] + a[index2];19         a[index2] = a[index1] - a[index2];20         a[index1] = a[index1] - a[index2];21     }22 23     private void print(int[] a){24         for(int i=0;i<a.length;i++){25             System.out.print(a[i]+" ");26         }27         System.out.println();28     }29 30     private void init_sort(int[] a,int n){        31         int m=(n+1)/2;    32         for(int i=0;i<m;i++){33             boolean flag=build_heap(a,n,i);34             //如果孩子之间有交换,就要重新开始35             if(flag){36                 i=-1;37             }38         }39         40             41     }42     //返回一个标记,如果有根与孩子交换就要重新从顶根开始查找不满足最大堆树结构43     private boolean build_heap(int a[],int n,int i){44         int l_child=2*i+1;//左孩子45         int r_child=2*i+2;//右孩子46         if(r_child>n){           //判断是否有右孩子,没有的话直接比较,小于右孩子则交换47             if(a[i]<a[l_child]){48                 swap(a, i, l_child);49                 return true;50             }else{51                     return false;52                 }53         }54         //在根与两个孩子之间找出最大的那个值进行交换55         if(a[i]<a[l_child]){56             if(a[l_child]>a[r_child]){57                 //交换根与左孩子的值58                 swap(a, i, l_child);59                 return true;60             }else{61                 //交换根与右孩子的值62                 swap(a, i, r_child);63                 return true;64             }65         }else if(a[i]<a[r_child]){66                 //交换根与右孩子的值67             swap(a, i, r_child);68                 return true;69         }70         return false;71             72     }73     public static void main(String[] args) 74     {75         Heap h=new Heap();76         int [] a={17,8,45,84,2,94};77         h.heap_sort(a,a.length-1);78     }79 }

 

堆排序