您的位置:首页 > 编程语言 > Java开发

堆排序的java实现

2016-03-04 09:57 603 查看
在堆排序的java实现中,待排序的数组arr[0]是空出来不使用的,因为java数组的下标是从0开始的,而堆的起始下标为1(保证满足父节点与左右节点的算数关系)。因此arr[1],arr[2]...ar[arr.length - 1]是被排序的元素集合。
在堆排序算法中,每次将堆的最后一个元素与堆顶元素(当前堆最大元素)进行交换,然后将堆大小heapSize减1,再对根节点进行最大堆的维护,依次类推。这里要注意的是数组的大小是不变的,而堆的大小是变化的,不能简单地认为heapSize=arr.length - 1.正因为此,在最大堆的维护算法和建堆算法中要传入堆大小这个参数。

package mec.lanqiao;

import java.util.Arrays;

public class Main {
	static int left(int i) {
		return i * 2;
	}

	static int right(int i) {
		return i * 2 + 1;
	}

	static int parent(int i) {
		return i / 2;
	}

	static void swap(int[] arr, int i, int j) {
		int temp = arr[i];
		arr[i] = arr[j];
		arr[j] = temp;
	}

	// 维护最大堆性质,假定以left(i),right(i)为根的二叉树都是最大堆
	static void maxHeapify(int[] arr, int i, int heapSize) {
		int l = left(i);
		int r = right(i);
		int largest = i;
		if (l <= heapSize && arr[l] > arr[largest])
			largest = l;
		if (r <= heapSize && arr[r] > arr[largest])
			largest = r;
		if (largest != i) {
			swap(arr, i, largest);
			maxHeapify(arr, largest, heapSize);
		}
	}

	// 建堆
	static void buildMaxHeap(int[] arr) {
		int heapSize = arr.length - 1;
		for (int i = (arr.length - 1) / 2; i > 0; i--) {
			maxHeapify(arr, i, heapSize);
		}
	}

	// 堆排序
	static void heapSort(int[] arr, int heapSize) {
		buildMaxHeap(arr);
		for (int i = arr.length - 1; i > 1; i--) {
			swap(arr, 1, i);
			heapSize--;
			maxHeapify(arr, 1, heapSize);
		}
	}

	public static void main(String[] args) {
		int[] arr = { 0, 7, 16, 10, 8, 14, 9, 3, 2, 1, 4 };
		// maxHeapify(arr, 1, 10);
		// [16, 14, 10, 8, 7, 9, 3, 2, 4, 1]
		heapSort(arr, arr.length - 1);
		System.out.println(Arrays.toString(arr));
	}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: