堆排序

最新推荐文章于 2023-09-21 13:21:58 发布

替这位空想家惊讶

最新推荐文章于 2023-09-21 13:21:58 发布

阅读量96

点赞数

分类专栏：算法模板

本文链接：https://blog.csdn.net/weixin_43708069/article/details/107618875

版权

算法模板专栏收录该内容

2 篇文章 0 订阅

订阅专栏

大顶堆正序排序

#include<bits/stdc++.h>
#define N 200 
using namespace std;

void heapify(int arr[], int pos, int n){
	int left_child = 2*pos, right_child = 2*pos+1;
	
	if(left_child <= n && arr[pos] < arr[left_child])
		swap(arr[pos], arr[left_child]);
		
	if(right_child <= n && arr[pos] < arr[right_child])
		swap(arr[pos], arr[right_child]);
} 

void sort(int arr[], int n){
	for(int i=n/2; i>=1; i--)
		heapify(arr, i, n);
		
	for(int i=n; i>1; i--){
		swap(arr[1], arr[i]);
		
		for(int j=(i-1)/2; j>=1; j--)
			heapify(arr, j, i-1);
	}
}

int main(){
	int n, arr[N];
	cout<<"请输入待排序数组大小(输入0退出)：";
	cin>>n;
	
	while(n){
		cout<<"待排序数组：";
		for(int i=1; i<=n; i++)
			cin>>arr[i];

		sort(arr, n);
		
		cout<<"排序后数组：";
		for(int i=1; i<=n; i++)
			cout<<arr[i]<<" ";
		cout<<endl<<endl<<"请输入待排序数组大小(输入0退出)：";
		cin>>n;
	}
	
	return 0;
}

小顶堆逆序排序

#include<bits/stdc++.h>
#define N 200 
using namespace std;

void heapify(int arr[], int pos, int n){
	int left_child = 2*pos, right_child = 2*pos+1;
	
	if(left_child <= n && arr[pos] > arr[left_child])
		swap(arr[pos], arr[left_child]);
		
	if(right_child <= n && arr[pos] > arr[right_child])
		swap(arr[pos], arr[right_child]);
} 

void sort(int arr[], int n){
	for(int i=n/2; i>=1; i--)
		heapify(arr, i, n);
		
	for(int i=n; i>1; i--){
		swap(arr[1], arr[i]);
		
		for(int j=(i-1)/2; j>=1; j--)
			heapify(arr, j, i-1);
	}
}

int main(){
	int n, arr[N];
	cout<<"请输入待排序数组大小(输入0退出)：";
	cin>>n;
	
	while(n){
		cout<<"待排序数组：";
		for(int i=1; i<=n; i++)
			cin>>arr[i];

		sort(arr, n);
		
		cout<<"排序后数组：";
		for(int i=1; i<=n; i++)
			cout<<arr[i]<<" ";
		cout<<endl<<endl<<"请输入待排序数组大小(输入0退出)：";
		cin>>n;
	}
	
	return 0;
}