C++ young 程序库——y_algorithm_heap.hpp

文件位置：young/algoithm/y_algorithm_heap.hpp

The young Library

Permission to use, copy, modify, distribute and sell this software for any

purpose is hereby granted without fee, provided that the above copyright

notice appear in all copies and that both that copyright notice and this

permission notice appear in supporting documentation.

The author make no representations about the suitability of this software

for any purpose. It is provided "as is" without express or implied warranty.

//-----------------------------------------------------------------------------

#ifndef __MACRO_CPLUSPLUS_YOUNG_LIBRARY_ALGORITHM_HEAP_HEADER_FILE__

#define __MACRO_CPLUSPLUS_YOUNG_LIBRARY_ALGORITHM_HEAP_HEADER_FILE__

//-----------------------------------------------------------------------------

#include "../y_iterator.hpp"

//-----------------------------------------------------------------------------

__MACRO_CPLUSPLUS_YOUNG_LIBRARY_BEGIN_NAMESPACE__

//-----------------------------------------------------------------------------

template< typename RandomAccessIterator >

inline

void push_heap( RandomAccessIterator first, RandomAccessIterator last )

{

typedef typename iterator_traits<RandomAccessIterator>::value_type

value_t;

typedef typename iterator_traits<RandomAccessIterator>::difference_type

diff_t;

push_heap_aux( first, diff_t((last - first) - 1),

diff_t(0), value_t(*(last - 1)) );

}

template< typename RandomAccessIterator, typename Integer, typename T >

void push_heap_aux( RandomAccessIterator first, Integer holeIndex,

Integer topIndex, T value )

{

Integer parent = (holeIndex - 1) / 2;

//没有至根节点并且新值大于父节点时，向上调整插入点索引和父节点索引

while( (holeIndex > topIndex) && (*(first + parent) < value) )

{

*(first + holeIndex) = *(first + parent);

holeIndex = parent;

parent = (holeIndex - 1) / 2;

}

*(first + holeIndex) = value;

}

template< typename RandomAccessIterator, typename StrictWeakOrdering >

inline

void push_heap( RandomAccessIterator first, RandomAccessIterator last,

StrictWeakOrdering comp )

{

typedef typename iterator_traits<RandomAccessIterator>::value_type

value_t;

typedef typename iterator_traits<RandomAccessIterator>::difference_type

diff_t;

push_heap_aux( first, diff_t((last - first) - 1),

diff_t(0), value_t(*(last - 1)), comp );

}

template< typename RandomAccessIterator, typename Integer, typename T,

typename StrictWeakOrdering >

void push_heap_aux( RandomAccessIterator first, Integer holeIndex,

Integer topIndex, T value, StrictWeakOrdering comp )

{

Integer parent = (holeIndex - 1) / 2;

while( (holeIndex > topIndex) && comp(*(first + parent), value) )

{

*(first + holeIndex) = *(first + parent);

holeIndex = parent;

parent = (holeIndex - 1) / 2;

}

*(first + holeIndex) = value;

}

//-----------------------------------------------------------------------------

template< typename RandomAccessIterator >

inline

void pop_heap( RandomAccessIterator first, RandomAccessIterator last )

{

typedef typename iterator_traits<RandomAccessIterator>::value_type

value_t;

typedef typename iterator_traits<RandomAccessIterator>::difference_type

diff_t;

value_t value = *(last - 1); //要调整的值

*(last - 1) = *first; //被弹出的值

adjust_heap( first, diff_t(0), diff_t(last - 1 - first), value );

}

template< typename RandomAccessIterator, typename Integer, typename T >

void adjust_heap( RandomAccessIterator first, Integer holeIndex,

Integer len, T value )

{

Integer topIndex = holeIndex;

Integer secondChild = 2 * holeIndex + 2;

while( secondChild < len ) //未越界

{

//左、右子节点哪边更大，则从哪边向下调整

if( *(first + secondChild) < *(first + secondChild - 1) )

--secondChild;

*(first + holeIndex) = *(first + secondChild);

//调整洞节点和右子节点

holeIndex = secondChild;

secondChild = 2 * (secondChild + 1);

}

//尾节点为左子节点

if( secondChild == len )

{

*(first + holeIndex) = *(first + secondChild - 1);

holeIndex = secondChild - 1;

}

push_heap_aux( first, holeIndex, topIndex, value );

}

template< typename RandomAccessIterator, typename StrictWeakOrdering >

inline

void pop_heap( RandomAccessIterator first, RandomAccessIterator last,

StrictWeakOrdering comp )

{

typedef typename iterator_traits<RandomAccessIterator>::value_type

value_t;

typedef typename iterator_traits<RandomAccessIterator>::difference_type

diff_t;

value_t value = *(last - 1);

*(last - 1) = *first;

adjust_heap( first, diff_t(0), diff_t(last - 1 - first),

value, comp );

}

template< typename RandomAccessIterator, typename Integer, typename T,

typename StrictWeakOrdering >

void adjust_heap( RandomAccessIterator first, Integer holeIndex,

Integer len, T value, StrictWeakOrdering comp )

{

Integer topIndex = holeIndex;

Integer secondChild = 2 * holeIndex + 2;

while( secondChild < len )

{

if( comp( *(first + secondChild), *(first + secondChild - 1) ) )

--secondChild;

*(first + holeIndex) = *(first + secondChild);

holeIndex = secondChild;

secondChild = 2 * (secondChild + 1);

}

if( secondChild == len )

{

*(first + holeIndex) = *(first + secondChild - 1);

holeIndex = secondChild - 1;

}

push_heap_aux( first, holeIndex, topIndex, value, comp );

}

//-----------------------------------------------------------------------------

template< typename RandomAccessIterator >

void make_heap( RandomAccessIterator first, RandomAccessIterator last )

{

typedef typename iterator_traits<RandomAccessIterator>::value_type

value_t;

typedef typename iterator_traits<RandomAccessIterator>::difference_type

diff_t;

if( last - first < 2 )

return;

diff_t len = last - first;

diff_t holeIndex = (len - 2) / 2; //最后一个非树叶节点的索引

while(1)

{

adjust_heap( first, holeIndex, len, value_t(*(first + holeIndex)) );

if( holeIndex == 0 )

return;

--holeIndex;

}

template< typename RandomAccessIterator, typename StrictWeakOrdering >

void make_heap( RandomAccessIterator first, RandomAccessIterator last,

StrictWeakOrdering comp )

{

typedef typename iterator_traits<RandomAccessIterator>::value_type

value_t;

typedef typename iterator_traits<RandomAccessIterator>::difference_type

diff_t;

if( last - first < 2 )

return;

diff_t len = last - first;

diff_t holeIndex = (len - 2) / 2;

while(1)

{

adjust_heap( first, holeIndex, len, value_t(*(first + holeIndex)), comp );

if( holeIndex == 0 )

return;

--holeIndex;

}

//-----------------------------------------------------------------------------

template< typename RandomAccessIterator >

void sort_heap( RandomAccessIterator first, RandomAccessIterator last )

{

while( (last - first) > 1 )

{

pop_heap( first, last );

--last;

}

template< typename RandomAccessIterator, typename StrictWeakOrdering >

void sort_heap( RandomAccessIterator first, RandomAccessIterator last,

StrictWeakOrdering comp )

{

while( (last - first) > 1 )

{

pop_heap( first, last, comp );

--last;

}

//-----------------------------------------------------------------------------

template< typename RandomAccessIterator >

bool is_heap( RandomAccessIterator first, RandomAccessIterator last )

{

typedef typename iterator_traits<RandomAccessIterator>::difference_type

diff_t;

diff_t len = last - first;

if( len < 2 )

return true;

diff_t parent = 0;

for( diff_t child = 1; child < len; ++child )

{

if( *(first + parent) < *(first + child) )

return false;

if( child % 2 == 0 )

++parent;

}

return true;

}

template< typename RandomAccessIterator, typename StrictWeakOrdering >

bool is_heap( RandomAccessIterator first, RandomAccessIterator last,

StrictWeakOrdering comp )

{

typedef typename iterator_traits<RandomAccessIterator>::difference_type

diff_t;

diff_t len = last - first;

if( len < 2 )

return true;

diff_t parent = 0;

for( diff_t child = 1; child < len; ++child )

{

if( comp(*(first + parent), *(first + child)) )

return false;

if( child % 2 == 0 )

++parent;

}

return true;

}

//-----------------------------------------------------------------------------

__MACRO_CPLUSPLUS_YOUNG_LIBRARY_END_NAMESPACE__

#endif

//-----------------------------------------------------------------------------