java堆排序

package com.zz.sort;

import java.util.Random;

public class HeapSort {
	

	/**
	 * @param args
	 */
	public static void main(String[] args) {
		int iLength = 5000;
		int[] iArgs = new int[iLength];
		for (int i = 0; i < iLength; i++) {
			Random objRandom = new Random();
			iArgs[i] = objRandom.nextInt(10000);
		}
		new HeapSort().heapSort(iArgs,0,iArgs.length);
		for (int i = 0; i < iArgs.length; i++) {
			System.out.print(iArgs[i] + " ");
		}

	}
	
	/**
	 * 數組排序成大根堆
	 * 
	 * @param iArgs 數組
	 * @param parent 父節點
	 * @param length 數組長度
	 */
	private void bigHeap(int[] iArgs,int parent,int length) {
		//父節點的值
		int iTempParent = iArgs[parent];
		//左節點的位置
		int child = 2*parent + 1;
		
		while( child < length) {
			
			//存在右子節點,並且右子結點比父節點的值大
			if(child + 1 < length && iArgs[child + 1] > iArgs[child]) {
				child ++;
			}
			//父節點的值比子節點的值大,程序直接跳出
			if(iArgs[parent] > iArgs[child]) {
				break;
			}
			
			//子節點比父節點大,把子節點的值賦給父節點
			iArgs[parent] = iArgs[child];
			iArgs[child] = iTempParent;
			
			//子節點當做父節點,重新排序
			parent = child; 
			child = 2* parent + 1;
		}
	}
	
	/**
	 * 大根堆最頂端節點和最右端尾端交換
	 * 
	 * @param iArgs 數組
	 * @param parent 父節點
	 * @param length 數組長度
	 */
	private void heapSort(int[] iArgs,int parent,int length ) {
		for ( int i = length/2 -1 ;i >= 0; i --) {
			bigHeap(iArgs,i,length);
		}
		
		for(int i = iArgs.length -1 ; i >= 0 ;i--) {
			//頂層父節點和尾端節點交換位置
			int temp = iArgs[0];
			iArgs[0] = iArgs[i];
			iArgs[i] = temp;
			
			//重構大根堆
			bigHeap(iArgs,0,i);
		}
		
	}
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章