堆排序的OC实现
Posted 小敏的博客
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了堆排序的OC实现相关的知识，希望对你有一定的参考价值。
/*
建议先看堆调整方法，堆调整了解了，整个排序算法就算掌握了
*/
- (void)viewDidLoad
{
[super viewDidLoad];
/*
测试数据
*/
NSArray *[email protected][@3,@2,@6,@4,@1,@0,@6,@7,@5];
NSMutableArray *mutable=[NSMutableArray arrayWithArray:array];
mutable=[self createMaxHeap:mutable];
NSInteger num=mutable.count;
/*
剩余的元素个数不为1时则继续调整，取出元素。取出的元素放在最后的一个节点。然后减小堆的元素的个数。所以大顶堆排序出来的是升序的。
*/
while (num>1)
{
[mutable exchangeObjectAtIndex:0 withObjectAtIndex:num-1];
mutable=[self maxHeapAdjust:mutable index:0 length:num-1];
num--;
}
}
/*
调整堆 递归调整的过程。这个调整堆的方法传入的是待调整的数组，数组元素的长度（为什么不直接用array.count呢？因为再进行排序
的时候，我们会动态更改无序堆的长度，而array的长度确是不变的，所以不用array.cout） 其实每调用一次调整堆方法，我们相当于只调整3个元素：父节点，左、右子节点。当左子结点是三者中最大的时候，把它和父节点进行交换。然后再递归调整以刚才的父节点（现在被降级为左子节点）为父节点的三个节点。此时为什么不用调整右子节点呢？这是由于我们建立大顶堆的过程中，都是自下而上进行调整的，此时我们没有动右子节点，且右子节点和现在的父节点（原来的左子节点）满足大顶堆的条件，所以不用调整。
*/
-(NSMutableArray*)maxHeapAdjust:(NSMutableArray *)array index:(NSInteger)index length:(NSInteger)length
{
//获取该节点的左子节点索引
NSInteger leftChildIndex =index*2+1;
//获取该节点的右子节点索引
NSInteger rightChildIndex=index*2+2;
//暂时把该索引当做最大值所对应的索引
NSInteger maxValueIndex=index;
// leftChildIndex<length
//array[leftChildIndex]>array[maxValueIndex]
判断左子节点的值是否大于当前最大值
if (leftChildIndex<length && array[leftChildIndex]>array[maxValueIndex])
{
//把左子节点的索引作为最大值所对应的索引
maxValueIndex=leftChildIndex;
}
/**
* rightChildIndex<length
* array[leftChildIndex] > array[maxValueIndex]
* 判断左子节点的值是否大于当前最大值
**/
if (rightChildIndex<length && array[rightChildIndex]>array[maxValueIndex])
{
maxValueIndex=rightChildIndex;
}
//如果该节点不是最大值所在的节点 则将其和最大值节点进行交换
if (maxValueIndex!=index)
{
[array exchangeObjectAtIndex:maxValueIndex withObjectAtIndex:index];
//递归乡下调整，此时maxValueIndex索引所对应的值是 刚才的父节点。
array=[self maxHeapAdjust:array index:maxValueIndex length:length];
}
return array;
}
-(NSMutableArray*)createMaxHeap:(NSMutableArray*)array
{
/*
从最后一个非叶子节点开始 自下而上进行调整堆
*/
for (NSInteger i=(array.count/2-1);i>=0; --i)
{
array=[self maxHeapAdjust:array index:i length:array.count] ;
}
return array;
}
以上是关于堆排序的OC实现的主要内容，如果未能解决你的问题，请参考以下文章