堆排序
思路
1.将数组构造成大顶堆,假设此时有n个节点,数组长度为n;
2.此时大顶堆的根节点是数组中最大的数,将其与数组末尾元素交换,将剩下的长度为n-1的数组堆化(注意:堆化是指在左右子树都为大顶堆的前提下,插入一个新的根节点,将新的堆变成大顶堆的过程,这和构造一个堆是不一样的)。
3.堆化之后得到大顶堆,其根节点值是剩下n-1个数中最大的,将其与数组末尾第2个元素交换,剩下长度为n-2的数组,然后将剩下的数组堆化,重复步骤2,直到剩下的数组长度为1.
此时数组已然有序。
#include<iostream>
#include<vector>
using namespace std;
//交换数组中给定两个位置值
void swap(vector<int> &tree,int x,int y)//
{
int temp = tree[x];
tree[x] = tree[y];
tree[y] = temp;
}
//打印数组
void ShowVec(const vector<int>& a )
{
for(int i=0;i<a.size();i++)
{
cout<<a[i]<<" ";
}
cout<<endl;
}
//堆化:假设在左右子树都为大顶堆的条件下,插入一个根节点构成大顶堆,位置i为根节点。
void Heapify(vector<int> &tree,int n,int i)
{
int c1 = 2*i+1;
int c2 = 2*i+2;
int parent = (i-1)/2;
int max =i;
if(c1<n&&tree[c1]>tree[max]) max =c1;
if(c2<n&&tree[c2]>tree[max]) max =c2;
if(max!=i)
{
swap(tree,i,max);
Heapify(tree,n,max);
}
}
// 将数组构造成大顶堆
void HeapBuild(vector<int>& a)
{
int n = a.size();
int lastParentNode = (n-2)/2;
for(int i = lastParentNode;i>=0;i--)
{
Heapify(a,n,i);
}
}
// 利用大顶堆进行排序
void HeapSort(vector<int>& tree)
{
int n = tree.size();
HeapBuild(tree);
for(int i=n-1;i>=0;i--)
{
swap(tree,0,i);
Heapify(tree,i,0);
}
}
int main()
{
int n;
cin>>n;
vector<int> tree;
for(int i =0;i<n;i++)
{
int temp;
cin>>temp;
tree.push_back(temp);
}
cout<<"输入的数组为:"<<endl;
ShowVec(tree);
HeapSort(tree);
cout<<"堆排序结果:"<<endl;
ShowVec(tree);
}

4689

被折叠的 条评论
为什么被折叠?



