|
def get_leaves(pre_order, leaves):
- i( Z6 u$ p% p if not pre_order:# E& Y1 s; x) R* n7 b. `9 o
return
' j: l$ _2 I! y3 K9 E
& X) i8 j7 Z1 m4 G8 g root = pre_order[0]
+ g; L# O4 A# c( ~9 c1 j if len(pre_order) == 1:) s u$ {) c _- {4 R0 ~% N
leaves.append(root)5 K) R1 u& C; O2 W. G E
return
9 t/ Z! V# d- |7 |* V; [ ( b2 C0 V. o+ R! i
left_start = None! w5 {& g( d5 Q: ^2 q
left_end = None) Y3 E% a0 y0 A3 ^: D
right_start = None
4 U5 d5 \& U5 P righ_end = None
" t- c% @/ D; ] for i in xrange(1, len(pre_order)):
8 w# i+ ~; ?. Q8 w; v0 w' k& e! B if not left_start and pre_order[i] <= root:
' d3 R7 y9 c! O! S/ m9 E left_start = i3 l" \0 }! X \/ D' H
) p, z6 E% b0 J: P% P if not right_start and pre_order[i] > root:' z A' g, w$ T' D% I; \7 u1 \8 O
right_start = i: `% A+ @2 Y$ Y+ k: g( u+ I
+ f5 a1 t% p0 c3 `" Z3 f$ M6 {. e* f
if left_start:
" u( d# t; E( U2 Z5 `3 k2 b# t5 ` if right_start: left_end = right_start- f+ J' u! f% a+ H9 o5 \* d
else: left_end = len(pre_order)' u( _6 Z3 I0 N5 }. l+ s5 o
get_leaves(pre_order[left_start:left_end], leaves)
E' l, K7 D: r, ^6 M' c m9 r r % R3 w% j5 Z( Q& k7 U0 e7 }
if right_start:: [: y. B5 d2 M7 U5 k
right_end = len(pre_order)( \, f B, F$ d9 {. z
get_leaves(pre_order[right_start:right_end], leaves)
6 x; G) D$ x' p- X
# [. V( N9 t* G% I, v" K( da = [5, 3, 2, 1, 4, 7, 6, 8]
6 b2 k% j; I# n) W6 r, Na_leaves =[]
Q' _5 d! w# A Yget_leaves(a, a_leaves)
! I1 E; u) C3 }- G) O" ?print a_leaves
| 
发表于 9-10-2018 08:38 AM
