堆排序

mac2022-06-30  30

void adjust(vector<int> &arr, int len, int index){ // 递归方式构建大根堆(len是arr的长度,index是第一个非叶子节点的下标) int left = 2*index+1;// index的左子节点 int right = 2*index+2;// index的右子节点 int maxIdx = index; if(left<len && arr[left]>arr[maxIdx]) maxIdx = left; if(right< len && arr[right]>arr[maxIdx]) maxIdx = right; if (maxIdx!=index) { swap(arr[maxIdx], arr[index]); adjust(arr, len, maxIdx); } } void heapsort(vector<int> &arr, int size){ for(int i = size/2; i>=0; --i){// 构建大根堆(从最后一个非叶子节点向上) adjust(arr, size, i); } for(int i = size-1;i>=1;--i){ swap(arr[0], arr[i]);// 将当前最大的放置到数组末尾 adjust(arr, i, 0);// 将未完成排序的部分继续进行堆排序 } } int main() { int n,m; while(cin>>n){ vector<int> arr(n); for(int i = 0; i<n; i++) { cin>>m; arr[i]=m; } heapsort(arr, arr.size()); for(int i = 0; i<arr.size();++i){ cout<<arr[i]<<endl; } } return 0; }

 

最新回复(0)