堆排序

想学会堆排序,就应该先理解堆结构。

堆分为大根堆和小根堆,大根堆就是所有的根节点都比他的子节点都要大,小根堆同理。

我们进行堆排序的时候,其实并不是真的要用到二叉树(完全二叉树),而只是借用这个知识来理解,可以将一个数组想象成一棵树。例如i节点的父节点就是(i+1)/2,i节点的左子节点就是i*2+1,右子节点就是i*2+2。

heapInsert:将数组的0到1位置的数变成一个大根堆,然后加入2位置的数,整体是一个大根堆,方法就是刚进来的i位置的数与他的父节点比较,如果大于父节点就交换,然后i位置变成了他的父节点的位置,然后再比较i位置与他的父节点大小,,,,如此类推。这样就形成了一整个大根堆。

heapfy:当大根堆的某一个位置 i 的数变小了,就需要对堆的位置进行从新调整,方法就是i位置的数与他的两个子节点中比较大的值交换,然后i也变成他的大的子节点的位置,然后再与子节点比较,如此类推,堆就调整好了。

了解了堆,就可以轻松的学会堆排序了,堆排序的步骤就是:将数组调整成为大根堆;这个时候头节点是最大值,将他与最后一个位置的元素交换,利用heapfy重新调整堆,然后heapSize(堆的有效范围0~heapSize)减一,这个时候,最大值就到了数组的最后一个位置,重复上面的操作,第二大的数就到了数组的倒数第二个位置了,,,,,一直到完成排序,也就是heapSize等于0的时候。

使用小根堆进行排序也是同样的道理,这里就不多讲了,得出来的是大到小序列

下面看代码(使用了以前讲过的对数器):

/**
 * 堆排
 */
import java.util.Arrays;
public class C02_HeapSort {
	public static void heapSort(int arr[]){
		if(arr==null || arr.length<2){
			return ;
		}
		//将0~i位置变成大根堆
		for (int i = 0; i < arr.length; i++) {
			heapInsert(arr,i);
		}
		int heapSize = arr.length;
		swap(arr, 0, --heapSize);
		while(heapSize>0){
			heapfy(arr, 0, heapSize);
			swap(arr, 0, --heapSize);
		}
	}

	//将0~i位置变成大根堆
	private static void heapInsert(int[] arr, int i) {
		while(arr[i]>arr[(i-1)/2]){
			swap(arr,i,(i-1)/2);
			i = (i-1)/2;
		}
	}

	//当堆中一个值(index)变小的时候,对堆进行调整
	//数组arr,index位置的值变小,size是堆的范围(0~size)
	private static void heapfy(int[] arr,int index,int heapSize){
		int left = index*2+1;//index的左孩子
		while(left<heapSize){//左孩子不越界
			//右孩子不越界并且左孩子大于右孩子的时候large是left
			int large = left+1<heapSize && arr[left+1]>arr[left] ? left+1 : left;
			large = arr[index]<arr[large]?large:index;
			if(index==large){
				break;
			}
			swap(arr, large, index);
			index = large;
			left = index*2+1;
		}
	}
	//交换
	private static void swap(int[] arr, int i, int j) {
		int t = arr[i];
		arr[i] = arr[j];
		arr[j] = t;
	}
	
	for test//
	// for test
	public static void comparator(int[] arr) {
		Arrays.sort(arr);
	}

	// for test
	public static int[] generateRandomArray(int maxSize, int maxValue) {
		int[] arr = new int[(int) ((maxSize + 1) * Math.random())];
		for (int i = 0; i < arr.length; i++) {
			arr[i] = (int) ((maxValue + 1) * Math.random()) - (int) (maxValue * Math.random());
		}
		return arr;
	}

	// for test
	public static int[] copyArray(int[] arr) {
		if (arr == null) {
			return null;
		}
		int[] res = new int[arr.length];
		for (int i = 0; i < arr.length; i++) {
			res[i] = arr[i];
		}
		return res;
	}

	// for test
	public static boolean isEqual(int[] arr1, int[] arr2) {
		if ((arr1 == null && arr2 != null) || (arr1 != null && arr2 == null)) {
			return false;
		}
		if (arr1 == null && arr2 == null) {
			return true;
		}
		if (arr1.length != arr2.length) {
			return false;
		}
		for (int i = 0; i < arr1.length; i++) {
			if (arr1[i] != arr2[i]) {
				return false;
			}
		}
		return true;
	}

	// for test
	public static void printArray(int[] arr) {
		if (arr == null) {
			return;
		}
		for (int i = 0; i < arr.length; i++) {
			System.out.print(arr[i] + " ");
		}
		System.out.println();
	}

	// for test
	public static void main(String[] args) {
		int testTime = 1000000;
		int maxSize = 100;
		int maxValue = 100;
		boolean succeed = true;
		for (int i = 0; i < testTime; i++) {
			int[] arr1 = generateRandomArray(maxSize, maxValue);
			int[] arr2 = copyArray(arr1);
			heapSort(arr1);
			comparator(arr2);
			if (!isEqual(arr1, arr2)) {
				succeed = false;
				break;
			}
		}
		System.out.println(succeed ? "Nice!" : "Fucking fucked!");

		int[] arr = generateRandomArray(maxSize, maxValue);
		printArray(arr);
		heapSort(arr);
		printArray(arr);
	}
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值