首先我们要了解二叉堆。
二叉堆的定义
堆是一个被完全填满的二叉树.
而二叉堆又满足一下两个特性:
1.父节点的值总是大于等于(或者小于等于)左儿子的值以及右儿子的值
2.二叉堆中,任一节点的左子树以及右子树(如果有的话)均为二叉堆(都是最大堆或者最小堆)。
如果父节点的值总是大于等于儿子节点的值,那么这个堆就称为最大堆。
如果父节点的值总是小于等于儿子节点的值,那么这个堆就称为最小堆。
二叉树例图:
由于完全二叉树的规律性,我们可以用数组来存储。
对于数组任一位置i上的元素,其左儿子的位置是2*i,其右儿子的位置是2*i+1,父亲的位置是i/2(如果根节点从数组下标0开始存储,那么父亲的位置为(i-1)/2)。
堆排序
堆排序大致可分为以下两个步骤:
1.初始化堆
2.排序
初始化堆:
public static void heapChange(int[] num, int start, int end)
{
int temp = num[start];
for(int i=2*start+1; i<=end; i = 2*i+1)
{
//找左右儿子中的最小值
if(i<end&&num[i+1]<num[i])
i++;
if(temp<=num[i])
break;
num[start] = num[i];
start = i;
}
num[start] = temp;
}
public initial(int[] num,int n)
{
//从n/2开始是因为n个数的堆其后一般为叶子节点,默认符合堆的性质
for(int i=n/2;i>=0;i--)
heapChange(num,i,n);
}
排序:
进行排序的基本思路是这样的,鉴于堆的根节点的值是整个堆中的最值(最小或者最大)这一特性,将根节点与最后一个节点对换,形成去除最值后的一个新堆(大小减了1),对新堆进行调整使得其符合上述特性,然后再进行类似操作直到此堆只剩下一个值。此时,得到的就是排序后的数组(用最大堆可获得升序数组,用最小堆可获得降序数组)。
for(int i=n;i>=1;i--)
{
//交换
int temp = num[i];
num[i] = num[0];
num[0] = temp;
heapChange(num,0,i-1);
}
测试代码:
package Heap;
public class heap {
public static void heapChange(int[] num, int start, int end)
{
int temp = num[start];
for(int i=2*start+1; i<=end; i = 2*i+1)
{
//找左右儿子中的最小值
if(i<end&&num[i+1]<num[i])
i++;
if(temp<=num[i])
break;
num[start] = num[i];
start = i;
}
num[start] = temp;
}
public static void heapSort(int[] num, int n)
{
//初始化堆,i从n/2开始是因为n个数形成的堆,其n/2个叶节点默认符合堆
for(int i=n/2;i>=0;i--)
{
heapChange(num,i,n);
}
//排序
for(int i=n;i>=1;i--)
{
//交换
int temp = num[i];
num[i] = num[0];
num[0] = temp;
heapChange(num,0,i-1);
}
}
public static void main(String[] arg)
{
int[] num = {2,3,1,5,4,8,5};
for(int temp:num)
System.out.print(temp);
System.out.println();
heapSort(num,num.length-1);
for(int temp:num)
System.out.print(temp);
System.out.println();
}
}
输出:
总之,堆排序就是循环抽取堆中的最值,将其存入数组。