关闭

【刷题之路】堆排序

标签: 刷题二叉树堆排序
110人阅读 评论(1) 收藏 举报

经典算法

class HeapSort {
public:
    int* heapSort(int* A, int n) {
        // write code her
        maxheap(A,n);
        for(int i=n-1;i>=0;i--){
            swap(A[0],A[i]); //大根堆堆顶元素为堆中最大的,将堆顶元素移到数组尾部,并且将最后一个元素移过来,重新调整(n-1)个数的大根堆
            fixdown(A,0,i);
        }
        return A;


    }
        void maxheap(int num[],int n) //建立大根堆,从倒数第一个父节点开始调整


            int i;
            for(i=n/2-1;i>=0;i--){
                fixdown(num,i,n-1);
            }
        }
        void fixdown(int nums[],int i,int n) //堆调整

{
            int j,temp;
            int flag=1;
            while(2*i+1<n && flag)

{
                if(2*i+2>n-1) j=2*i+1; //首先选择头结点的两个子节点中较大的一个
                else if(nums[2*i+1]<nums[2*i+2] && i*2+2< n) j=2*i+2;
                else j=2*i+1;
                if(nums[i]<nums[j]) //如果头结点小于较大节点,则交换位置,因为之前的位置是调整好的,所以只需要调整num[i]即可,重复操作直到新加入节点下沉到一个合适的位置

{
                    temp=nums[i];
                    nums[i]=nums[j];
                    nums[j]=temp;
                    i=j;
                }
                else flag=0;
            }
  
        }
};

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:11744次
    • 积分:811
    • 等级:
    • 排名:千里之外
    • 原创:72篇
    • 转载:2篇
    • 译文:0篇
    • 评论:1条