void heapify(vector<int>& nums, int n, int i)//对有⼀定顺序的堆,
//当前第i个结点取根左右的最⼤值(这个操作称heapfiy)
{
int l = i * 2 + 1, r = i * 2 + 2;
int max = i;
if (l<n && nums[l]>nums[max])
max = l;
if (r<n && nums[r]>nums[max])
max = r;
if (max != i)
{
swap(nums[max], nums[i]);
heapify(nums, n, max);
}
}
void heapify_build(vector<int>& nums, int n)//建⽴⼤根堆,从树的倒数第⼆层最后⼀个有子节点的结点开始,
//对每个结点进⾏heapify操作,然后向上⾛
{
int temp = (n - 1) / 2;
for (int i = temp; i >= 0; i--)
heapify(nums, n, i);
for (int i = 0; i < nums.size(); i++)
cout << nums[i] << " ";
cout << endl;
}
void heapify_sort(vector<int>& nums, int n)//建⽴⼤根堆之后,每次交换最后⼀个结点和根节点(最⼤值),
//对交换后的根节点继续进⾏heapify(此时堆的最后⼀位是最⼤值,因此不⽤管他,n变为n-1)
{
heapify_build(nums, n);
for (int i = 0; i < n; i++)
{
swap(nums.front(), nums[n - i - 1]);
heapify(nums, n - i - 1, 0);
}
}
堆排序模板
最新推荐文章于 2024-12-19 20:08:13 发布