a
XC?h]Y @ s@ d Z g dZdd 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 ey Y n0 ed(kr|d d? }| | }||k r>|| |< |}qq>q|| |< d S Nr
r r startposposnewitem parentposparentr 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* childposrightposr 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" NFkeyreversec 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 |dkrz2|d \} } } }
| V | |
d< |||
qW q t y || Y q0 q|r|d \} } } | V | j
E d H d S ttt |D ]H\} }
z(|
j
} | } || | | | | | g W n t yd Y n0 q || t |dkrzF|d \}} } } }
| V | } | | |
d< | |
d< |||
qW n t y || Y n0 qr|r|d \}} } } | V | j
E d H d S )Nr
r r )r r% r# r$ r r r enumeratemapiter__next__
StopIterationr
__self__)r4 r5 iterableshh_append_heapify_heappop _heapreplace directionorderitnextvalues 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| dd | S d u rt |}dd tt| |D }|s|S t | |d d }| }t
} |D ].}
|
|k r| ||
|f |d \}} |d7 }q| dd |D S t |} fddtt| |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 | ]\}}||fqS 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 | ]\}} |||fqS r r rM rL r r rP rQ c S s g | ]\}}}|qS r r rN krD rO r r r rP rQ ) r9 objectminr
TypeErrorAttributeErrorsortedzipr r% r$ sortr iterabler4 rE sentinelresultsizetoprD 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| ddd | 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 dttd| 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 | ]\}}||fqS 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 | ]\}} |||fqS 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__ZdoctestprintZtestmodr r r r ! sH ^
5
<
;