zoukankan      html  css  js  c++  java
  • 算法:堆排序

    # 堆排序的Python实现
    #
    coding:utf-8 import random import time t = time.time() # 堆排序, 按照从小到大排序,所以建立大根堆 # 列表的第一个元素没有使用 def ShiftDown(a, i): if not a: return j = i N = len(a) while 2*j < N: # 存在左孩子时 ci = 2 * j # 左孩子节点下标 if ci + 1 < N and a[ci] < a[ci + 1]: # ci保存左右孩子节点中大值的下标 ci += 1 if a[j] < a[ci]: # 和孩子互换,否则,该元素位置正确,可以退出 a[j], a[ci] = a[ci], a[j] else: break j = ci def BuildHeap(a): if not a: return i = len(a)/2 while i >= 1: #print a[i] ShiftDown(a, i) i -= 1 def HeapSort(a): BuildHeap(a) b = [] while len(a) > 1: # 列表的第一个元素没有使用 a[1], a[-1] = a[-1], a[1] # 将最大元素和最后一个元素互换,弹出最大元素,重新建造堆 b.append(a.pop()) BuildHeap(a) a = b[::-1] b.sort() print a == b #print b[::-1] #a[N - 1], a[0] = a[0], a[N - 1] a = [random.randint(1,100) for i in range(1001)] #a = [60, 39, 80, 25, 66, 3, 93, 92, 11, 11, 38] print a BuildHeap(a) print a HeapSort(a) print time.time() - t

    以下是C语言实现:

     1 // 最大堆实现, 数组下标从1开始,a[0]不使用。
     2 
     3 // 交换两数
     4 void swap(int &a, int &b) {
     5     int t = a;
     6     a = b;
     7     b = t;
     8 }
     9 
    10 // 把第i个元素向上移动
    11 void ShiftUp(int a[], int i) {
    12     while(i>1 && a[i]>a[i/2]) {
    13         swap(a[i], a[i/2]);
    14         i >>= 1;
    15     }
    16 }
    17 
    18 // 把第i个元素向下移动
    19 void ShiftDown(int a[], int n, int i) {
    20     while((i=2*i) <= n) {
    21         if(i+1<=n && a[i+1]>a[i]) ++i;
    22         if(a[i] > a[i/2]) swap(a[i], a[i/2]);
    23         else break;
    24     }
    25 }
    26 
    27 // 把数组a变成具备最大堆性质的数组
    28 void MakeHeap(int a[], int n) {
    29     for(int i=n/2; i>0; --i)
    30         ShiftDown(a, n, i);
    31 }
    32 
    33 // 向堆中插入元素x
    34 void Insert(int a[], int &n, int x) {
    35     a[++n] = x;
    36     ShiftUp(a, n);
    37 }
    38 
    39 // 删除堆中第i个元素
    40 void Del(int a[], int &n, int i) {
    41     a[i] = a[n--];
    42     if(i>1 && a[i]>a[i/2]) ShiftUp(a, i);
    43     else ShiftDown(a, n, i);
    44 }
    45 
    46 // 堆排序,时间复杂度O(nlogn)
    47 void HeapSort(int a[], int n) {
    48     MakeHeap(a, n);
    49     for(int i=n; i>1; --i) {
    50         swap(a[i], a[1]);
    51         ShiftDown(a, i-1, 1);
    52     }
    53 }

    还有另外一个版本: 

     1 //删除最大的元素
     2 int deletemax()
     3 {
     4     int t;
     5     t=h[ 1];//用一个临时变量记录堆顶点的值
     6     h[ 1]=h[ n];//将堆得最后一个点赋值到堆顶
     7     n--;//堆的元素减少1
     8     siftdown(1);//向下调整
     9     return t;//返回之前记录的堆得顶点的最大值
    10 }
    11 
    12 
    13 #include <stdio.h>
    14 int h[ 101];//用来存放堆的数组
    15 int n;//用来存储堆中元素的个数,也就是堆的大小
    16 
    17 //交换函数,用来交换堆中的两个元素的值
    18 void swap(int x,int y)
    19 {
    20     int t;
    21     t=h[ x];
    22     h[ x]=h[ y];
    23     h[ y]=t;
    24 }
    25   
    26 //向下调整函数
    27 void siftdown(int i) //传入一个需要向下调整的结点编号i,这里传入1,即从堆的顶点开始向下调整
    28 {
    29     int t,flag=0;//flag用来标记是否需要继续向下调整
    30     //当i结点有儿子的时候(其实是至少有左儿子的情况下)并且有需要继续调整的时候循环窒执行
    31     while( i*2<=n && flag==0 )
    32     {        
    33         //首先判断他和他左儿子的关系,并用t记录值较小的结点编号
    34         if( h[ i] > h[ i*2] )
    35             t=i*2;
    36         else
    37             t=i;
    38         //如果他有右儿子的情况下,再对右儿子进行讨论
    39         if(i*2+1 <= n)
    40         {
    41             //如果右儿子的值更小,更新较小的结点编号  
    42             if(h[ t] > h[ i*2+1])
    43                 t=i*2+1;
    44         }
    45         //如果发现最小的结点编号不是自己,说明子结点中有比父结点更小的  
    46         if(t!=i)
    47         {
    48             swap(t,i);//交换它们,注意swap函数需要自己来写
    49             i=t;//更新i为刚才与它交换的儿子结点的编号,便于接下来继续向下调整
    50         }
    51         else
    52             flag=1;//则否说明当前的父结点已经比两个子结点都要小了,不需要在进行调整了
    53     }
    54 }
    55   
    56 //建立堆的函数
    57 void creat()
    58 {
    59     int i;
    60     //从最后一个非叶结点到第1个结点依次进行向上调整
    61     for(i=n/2;i>=1;i--)
    62     {
    63         siftdown(i);
    64     }  
    65 }
    66   
    67 //删除最大的元素
    68 int deletemax()
    69 {
    70     int t;
    71     t=h[ 1];//用一个临时变量记录堆顶点的值
    72     h[ 1]=h[ n];//将堆得最后一个点赋值到堆顶
    73     n--;//堆的元素减少1
    74     siftdown(1);//向下调整
    75     return t;//返回之前记录的堆得顶点的最大值
    76 }
    77   
    78 int main()
    79 {
    80     int i,num;
    81     //读入数的个数
    82     scanf("%d",&num);
    83   
    84     for(i=1;i<=num;i++)
    85         scanf("%d",&h[ i]);
    86     n=num;   
    87   
    88     //建堆
    89     creat();
    90   
    91   
    92     //删除顶部元素,连续删除n次,其实夜就是从大到小把数输出来
    93     for(i=1;i<=num;i++)
    94         printf("%d ",deletemax());
    95   
    96     getchar();
    97     getchar();
    98     return 0;
    99 }

    参考文献:

    http://www.kancloud.cn/digest/make-thiner-programming-pearls/56357

    http://ahalei.blog.51cto.com/4767671/1427156

  • 相关阅读:
    cocos2d中sprite动画接口及动画实现思路总结
    iOS开发:小技巧积累
    cocos2d使用定时器
    Eclipse报错:cannot connect to VM
    Myeclipse console 端没有 tomcat 启动的log信息
    MyEclipse Web项目不能自动编译
    在MyEclipse中如何查看Spring/Hibernate/Struts/JDK等源码的方法
    【KMS】查看Windows的环境变量
    【KMS】如何预防用户多次点击提交按钮
    MyEclipse中选择一行的快捷键
  • 原文地址:https://www.cnblogs.com/bitpeng/p/4770080.html
Copyright © 2011-2022 走看看