Sorting Algorithm-Heap Sort

最新推荐文章于 2024-03-03 11:44:37 发布

Dufre.WC

最新推荐文章于 2024-03-03 11:44:37 发布

阅读量662

点赞数

文章标签：堆排序

本文链接：https://blog.csdn.net/u011391629/article/details/52421345

版权

Heap Sort-堆排序

Algorithm：

堆排序就是利用堆（假设是大顶堆）进行排序的方法。它的基本思想是，将待排序的序列构造成一个大顶堆。此时，整个序列的最大值就是堆顶的根结点。将它移走（其实就是将其与堆数组的末尾元素交换，此时末尾元素就是最大值），然后将剩余的n-1个序列重新构造成一个堆，这样就会得到n个元素的次大值。如此反复执行，便能得到一个有序序列了。

Ex：大顶堆定义（所有结点都大于等于其孩子）

重复上面的过程，就可以排好序了。

在程序中，需要解决两个问题

1、如何构造大顶堆

2、堆顶元素和最后一个元素交换后，如何调整剩下的元素成为一个新的堆。

Code：

void Heap_Adjust(vector<int> &v, int i,int len)   //len为调整的长度
{
	int lchild = 2 * i + 1;         //左孩子节点
	int rchild = 2 * i + 2;         //右孩子节点
	int temp = i;
	int key = v[i];
	if (i < (len / 2))  //不是叶节点
	{
		if (lchild < len&&v[lchild] > v[temp])  //必须是v[temp]
			temp = lchild;
		if (rchild < len&&v[rchild] > v[temp])
			temp = rchild;
		if (temp != i)
		{
			v[i] = v[temp];
			v[temp] = key;
			Heap_Adjust(v, temp,len);  //避免调整后temp节点不是堆
		}
	}
}
void Heap_Sort(vector<int> &v)
{
	//Build Heap
	for (int i = v.size() / 2-1; i >= 0; i--)
		Heap_Adjust(v, i,v.size());
	//调整
	for (int k = v.size() - 1; k >= 0; k--)
	{
		int key = v[0];
		v[0] = v[k];
		v[k] = key;
		Heap_Adjust(v, 0,k);
	}
}

非递归，这里只写调整堆得代码：

void Adjust(vector<int> &A,int i,int n){
        int l=2*i+1;
        int r=2*i+2;
        while(l<n || r<n){
            int cur=i;
            if(l<n && A[l]<A[cur])
                cur=l;
            if(r<n && A[r]<A[cur])
                cur=r;
            if(cur != i){
                swap(A[cur],A[i]);
                l=cur*2+1;
                r=cur*2+2;
                i=cur;
            }   
            else
                break;
        }
    }