heap.cpp
上传用户:yhdzpy8989
上传日期:2007-06-13
资源大小:13604k
文件大小:5k
源码类别:

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: heap.cpp,v $
  4.  * PRODUCTION Revision 1000.1  2004/06/01 18:10:06  gouriano
  5.  * PRODUCTION PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.2
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: heap.cpp,v 1000.1 2004/06/01 18:10:06 gouriano Exp $
  10. * ===========================================================================
  11. *
  12. *                            PUBLIC DOMAIN NOTICE
  13. *               National Center for Biotechnology Information
  14. *
  15. *  This software/database is a "United States Government Work" under the
  16. *  terms of the United States Copyright Act.  It was written as part of
  17. *  the author's official duties as a United States Government employee and
  18. *  thus cannot be copyrighted.  This software/database is freely available
  19. *  to the public for use. The National Library of Medicine and the U.S.
  20. *  Government have not placed any restriction on its use or reproduction.
  21. *
  22. *  Although all reasonable efforts have been taken to ensure the accuracy
  23. *  and reliability of the software and data, the NLM and the U.S.
  24. *  Government do not and cannot warrant the performance or results that
  25. *  may be obtained by using this software or data. The NLM and the U.S.
  26. *  Government disclaim all warranties, express or implied, including
  27. *  warranties of performance, merchantability or fitness for any particular
  28. *  purpose.
  29. *
  30. *  Please cite the author in any work or product based on this material.
  31. *
  32. * ===========================================================================
  33. *
  34. * Author:  Richard Desper
  35. *
  36. * File Description:  heap.cpp
  37. *
  38. *    A part of the Miminum Evolution algorithm
  39. *
  40. */
  41. #include <ncbi_pch.hpp>
  42. #include <stdio.h>
  43. #include <stdlib.h>
  44. #include <math.h>
  45. #include "graph.h"
  46. BEGIN_NCBI_SCOPE
  47. BEGIN_SCOPE(fastme)
  48. typedef char boolean;
  49. int *initPerm(int size)
  50. {
  51.   int *p;
  52.   int i;
  53.   p = (int *) malloc(size*sizeof(int));
  54.   for(i = 0;i<size;i++)
  55.     p[i] = i;
  56.   return(p);
  57. }
  58. void permInverse(int *p, int *q, int length)
  59. {
  60.   int i;
  61.   for(i=0;i<length;i++)
  62.     q[p[i]] = i;
  63. }
  64. /*swaps two values of a permutation*/
  65. void meSwap(int *p, int *q, int i, int j)
  66. {
  67.   int temp;
  68.   temp = p[i];
  69.   p[i] = p[j];
  70.   p[j] = temp;
  71.   q[p[i]] = i;
  72.   q[p[j]] = j;
  73. }
  74. /*The usual Heapify function, tailored for our use with a heap of scores*/
  75. /*will use array p to keep track of indexes*/
  76. /*after scoreHeapify is called, the submeTree rooted at i 
  77.   will be a heap*/
  78. /*p goes from heap to array, q goes from array to heap*/
  79. void heapify(int *p, int *q, double *HeapArray, int i, int n)
  80. {
  81.   boolean moreswap = TRUE_FASTME;
  82.   do {
  83.     int left = 2 * i;
  84.     int right = 2* i + 1;
  85.     int smallest;
  86.     if ((left <= n) && (HeapArray[p[left]] < HeapArray[p[i]]))
  87.       smallest = left;
  88.     else
  89.       smallest = i;
  90.     if ((right <= n) && (HeapArray[p[right]] < HeapArray[p[smallest]]))
  91.       smallest = right;
  92.     if (smallest != i){
  93.       meSwap(p,q,i, smallest);     
  94.       /*push smallest up the heap*/    
  95.       i = smallest;            /*check next level down*/
  96.     }
  97.     else
  98.       moreswap = FALSE_FASTME;
  99.   } while(moreswap);
  100. }
  101. /*heap is of indices of elements of v, 
  102.   popHeap takes the index at position i and pushes it out of the heap
  103.   (by pushing it to the bottom of the heap, where it is not noticed)*/
  104. void reHeapElement(int *p, int *q, double *v, int length, int i)
  105. {
  106.   int up, here;
  107.   here = i;
  108.   up = i / 2;
  109.   if ((up > 0) && (v[p[here]] < v[p[up]]))
  110.     while ((up > 0) && (v[p[here]] < v[p[up]])) /*we push the new
  111.   value up the heap*/
  112.       {
  113. meSwap(p,q,up,here);
  114. here = up;
  115. up = here / 2;
  116.       }
  117.   else
  118.     heapify(p,q,v,i,length);
  119. }
  120. void popHeap(int *p, int *q, double *v, int length, int i)
  121. {
  122.   meSwap(p,q,i,length); /*puts new value at the last position in the heap*/
  123.   reHeapElement(p,q, v,length-1,i); /*put the swapped guy in the right place*/
  124. }
  125. void pushHeap(int *p, int *q, double *v, int length, int i)
  126. {
  127.   meSwap(p,q,i,length+1); /*puts new value at the last position in the heap*/
  128.   reHeapElement(p,q, v,length+1,length+1); /*put that guy in the right place*/
  129. }
  130. int makeThreshHeap(int *p, int *q, double *v, int arraySize, double thresh)
  131. {
  132.   int i, heapsize;
  133.   heapsize = 0;
  134.   for(i = 1; i < arraySize;i++)
  135.     if(v[q[i]] < thresh)
  136.       pushHeap(p,q,v,heapsize++,i);
  137.   return(heapsize);
  138. }
  139. END_SCOPE(fastme)
  140. END_NCBI_SCOPE
  141. /*
  142.  * ===========================================================================
  143.  * $Log: heap.cpp,v $
  144.  * Revision 1000.1  2004/06/01 18:10:06  gouriano
  145.  * PRODUCTION: UPGRADED [GCC34_MSVC7] Dev-tree R1.2
  146.  *
  147.  * Revision 1.2  2004/05/21 21:41:04  gorelenk
  148.  * Added PCH ncbi_pch.hpp
  149.  *
  150.  * Revision 1.1  2004/02/10 15:16:03  jcherry
  151.  * Initial version
  152.  *
  153.  * ===========================================================================
  154.  */