堆排序

大顶堆 正序排序

#include
#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
#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;
}