a
XC?h]Y � @ s@ d Z g d�Zdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
ddd �d d �Zd)d d �Zd*d!d"�Zz d#d$lT W n ey� Y n0 zd#d%lmZ W n ey� Y n0 zd#d&lm Z W n ey� Y n0 zd#d'lmZ W n e�y Y n0 ed(k�r|d d? }| | }||k r>|| |< |}qq>q|| |< d S �Nr
r �r �startpos�pos�newitem� parentpos�parentr r r r � s r c C s� t | �}|}| | }d| d }||k rj|d }||k rL| | | | k sL|}| | | |< |}d| d }q || |< t| ||� d S �Nr r
)r
r �r r) �endposr( r* �childpos�rightposr r r r s r c C sJ | | }||kr>|d d? }| | }||k r>|| |< |}qq>q|| |< d S r& r r' r r r �
_siftdown_max s r2 c C s� t | �}|}| | }d| d }||k rj|d }||k rL| | | | k sL|}| | | |< |}d| d }q || |< t| ||� d S r- )r
r2 r. r r r r" % s r" NF��key�reversec g s g }|j }|r t}t}t}d}nt}t}t}d}| d u �rttt |��D ]:\} }
z |
j
} || � | | | g� W qH t y� Y qH0 qH||� t |�dkr�z2|d \} } } }
| V | � |
d< |||
� q�W q� t y� ||� Y q�0 q�|�r|d \} } } | V | j
E d H d S ttt |��D ]H\} }
z(|
j
} | � } || | �| | | | g� W n t �yd Y n0 �q ||� t |�dk�r�zF|d \}} } } }
| V | � } | | �|
d< | |
d< |||
� �q�W n t �y� ||� Y n0 �qr|�r|d \}} } } | V | j
E d H d S )N�����r
r r )r r% r# r$ r r r � enumerate�map�iter�__next__�
StopIterationr
�__self__)r4 r5 � iterables�h�h_append�_heapify�_heappop� _heapreplace� direction�order�it�next�value�s� key_valuer r r r : sl
r c s� | dkr6t |�}t� }t||� d�}||u r0g S |gS z t|�}W n ttfyX Y n 0 | |krvt|� d�d | � S � d u �rt |�}dd� tt| �|�D �}|s�|S t |� |d d }| }t
} |D ].}
|
|k r�| ||
|f� |d \}} |d7 }q�|� � dd� |D �S t |�}� fdd�tt| �|�D �}|�s<|S t |� |d d }| }t
} |D ]>}
� |
�} | |k �r\| || ||
f� |d \}} }
|d7 }�q\|� � d d� |D �S )
Nr
��defaultr4 �r4 c S s g | ]\}}||f�qS r r ��.0r! �elemr r r �
� � z nsmallest..r c S s g | ] \}}|�qS r r �rN rO rD r r r rP � rQ c s g | ]\}}� |�||f�qS r r rM rL r r rP � rQ c S s g | ]\}}}|�qS r r �rN �krD rO r r r rP rQ ) r9 �object�minr
� TypeError�AttributeError�sorted�zipr r% r$ �sort�r �iterabler4 rE �sentinel�result�size�toprD rB rO �_orderrT �_elemr rL r r � sV
r c s� | dkr6t |�}t� }t||� d�}||u r0g S |gS z t|�}W n ttfyX Y n 0 | |krxt|� dd�d | � S � d u �r t |�}dd� ttd| d�|�D �}|s�|S t |� |d d }| }t
} |D ].}
||
k r�| ||
|f� |d \}} |d8 }q�|j dd � d
d� |D �S t |�}� fd d�ttd| d�|�D �}|�sP|S t |� |d d }| }t
} |D ]>}
� |
�} || k �rr| || ||
f� |d \}} }
|d8 }�qr|j dd � d d� |D �S )
Nr
rJ Tr3 c S s g | ]\}}||f�qS r r rM r r r rP " rQ z nlargest..r r6 )r5 c S s g | ] \}}|�qS r r rR r r r rP / rQ c s g | ]\}}� |�||f�qS r r rM rL r r rP 3 rQ c S s g | ]\}}}|�qS r r rS r r r rP A rQ ) r9 rU �maxr
rW rX rY rZ r r r r[ r\ r rL r r sV
"
r r )�*)r$ )r% )r# �__main__)N)N)� __about__�__all__r r r r r r# r$ r% r r r2 r" r r r �_heapq� ImportError�__name__Zdoctest�printZtestmodr r r r �! sH ^
5
<
;