c++ 堆排序 源代碼

#include "stdafx.h"
#include <iostream>
using namespace std;
template<typename T>
void adjust(T* arr,int sign,int len){
    T temp = arr[sign];
    //每一次循環都更新該父節點爲根的完全二叉樹最大堆
    for (int i = sign * 2 + 1; i < len; i = i * 2 + 1){
        if (i + 1 < len && arr[i + 1] > arr[i])
            i++;
        //判斷子節點 大於父節點 
        if (arr[i] > temp){
            arr[sign] = arr[i];
            sign = i;
        }
    }
    arr[sign] = temp;
}

template<typename T>
void swap(T* arr, int x, int y){
    T temp = arr[x];
    arr[x] = arr[y];
    arr[y] = temp;
}

template<typename T>
void sort(T* arr,int length){
    //1.從所有非葉子節點 構建初始大頂堆
    for (int i = length / 2 - 1; i >= 0; i--){
        adjust(arr, i, length);
    }
    //
    for (int i = length - 1; i; i--){
        //2.交換最大堆 和 相對的最後一個元素
        swap(arr, i, 0);
        //3.重新調整堆結構
        adjust(arr, 0, i);
    }
}

template<typename T>
void print(const T* ar,int len){
    for (int j = 0; j < len; j++){
        cout << j << " :  " << ar[j] << endl;
    }
}
int _tmain(int argc, _TCHAR* argv[])
{
    int arr[] = { 0,2,1,4,3,6,8,9,7,5};
    sort(arr, 10);
    print(arr, 10);
    getchar();
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章