算法导论示例-HeapSort

原创 2006年06月04日 00:00:00
/**
 * Introduction to Algorithms, Second Edition 
 * 6.4 HEAPSORT
 * @author 土豆爸爸
 * 
 */
public class HeapSort {
    int[] array;

    int heapSize;

    public void sort(int[] array) {
        this.array = array;
        buildMaxHeap();
        for (int i = array.length - 1; i > 0; i--) {
            //将最大值array[0]放在最后
            int temp = array[0];
            array[0] = array[i];
            array[i] = temp;
            heapSize--;
            //重排,把最大值放在array[0]
            maxHeapify(0);
        }
    }

    /**
     * 构建max heap
     * 
     */
    private void buildMaxHeap() {
        heapSize = array.length;
        for (int i = array.length / 2; i >= 0; i--) {
            maxHeapify(i);
        }
    }

    /**
     * 使数组的第i个元素按max heap规则重排
     * @param i
     *            元素索引
     */
    private void maxHeapify(int i) {
        int l = left(i);
        int r = right(i);
        int largest; // 当前结点/左子结点/右子结点中最大值的索引
        if (l < heapSize && array[l] > array[i]) {
            largest = l;
        } else {
            largest = i;
        }

        if (r < heapSize && array[r] > array[largest]) {
            largest = r;
        }

        if (largest != i) {
            // 如果最大值不是当前结点,进行交换
            int temp = array[i];
            array[i] = array[largest];
            array[largest] = temp;
            // 递归调用,直到当前结点比其子结点大
            maxHeapify(largest);
        }

    }

    /**
     * 计算结点索引为i的元素的左子结点的索引
     * @param i
     *            当前索引
     * @return 左子结点的索引
     */
    private int left(int i) {
        return 2 * i + 1;
    }

    /**
     * 计算结点索引为i的元素的右子结点的索引
     * @param i
     *            当前索引
     * @return 右子结点的索引
     */
    private int right(int i) {
        return 2 * i + 2;
    }
}

import junit.framework.TestCase;

public class HeapSortTest extends TestCase {
    public void testFixArray() {
        int[] array = { 5, 2, 4, 6, 1, 3 };
        HeapSort heapSort = new HeapSort();
        heapSort.sort(array);
        assertTrue(verifyOrdered(array));
    }

    public void testRandomArray() {
        int size = (int)(Math.random()* 1000);
        int[] array = new int[size];
        for(int i = 0; i < size; i++) {
            array[i] =  (int)(Math.random()* 10000);
        }
        
        HeapSort heapSort = new HeapSort();
        heapSort.sort(array);
        assertTrue(verifyOrdered(array));
    }
    
    private boolean verifyOrdered(int[] array) {
        for (int i = 1; i < array.length; i++) {
            if (array[i - 1] > array[i]) {
                return false;
            }
        }
        return true;
    }
}

相关文章推荐

算法导论------堆排序heapsort

目录 1.堆排序概述 2.二叉堆 3.二叉堆的存储 4.维护堆的性质 5.建堆操作 6.堆排序 7.C代码实现 8.参考资料 1.堆排序概述  简单选择排序是:假设排序序列为L[1...n]L[1.....

【算法导论学习-013】堆排序(Heapsort)

堆排序

链式哈希表(Hash Table)--算法导论示例

In computer science, a hash table is an associative array data structure that associates keys with v...

算法导论第三版习题5.4

5.4-1我的生日是一年中已经固定的一天,我们假设当有kk个人时,有人和我生日相同的概率为1/21/2,那么这kk个人中每一个人生日和我的不相同(即生日不在固定的某一天)的概率都为1−1/3651-1...
  • obguy
  • obguy
  • 2016年02月24日 20:31
  • 807

算法导论习题8-3—排序不同长度的数据项

题目: a)给定一个整数数组,其中不同的整数中包含的数字个数可能不同,但是该数组中,所有整数中总的数字数为n。说明如何在O(n)时间内对该数组进行排序 b)给定一个字符串数组,其中不同的串包含的字...

MIT:算法导论——7.1.基本数据结构_栈、队列、链表、有根树

【教材:第10章 基本数据结构】 (1)栈——后进先出 说明:采用一个数组S[1...n]来实现一个最多容纳n个元素的栈。  S.top=0时,栈不包含任何元素,栈空。下溢,上溢:S.top超过...

【算法导论】二叉查找树的操作C++实现

本代码为算法导论第12章中,伪代码的C++实现: #include #include using namespace std; /*二叉查找树结构*/ typedef struct BSTree...

算法导论CLRS 7.1 快速排序

/*7.1 快速排序 *QUICK-SORT */ #include #include #include #include using namespace std; typedef vector...

【算法导论】 第十二课 跳跃表

本节课介绍了一种全新的数据结构——跳跃表 跳跃表是一种简单又有趣的动态搜索数据结构,其主要优点在于其易于实现,而且很好的保证了其具有高效的性能,即2*O(lgn)的搜索性能 在此之...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:算法导论示例-HeapSort
举报原因:
原因补充:

(最多只允许输入30个字)