
将一棵二叉树按照前序遍历拆解成为一个假链表
。所谓的假链表是说,用二叉树的 right 指针,来表示链表中的 next 指针。
注意事项
不要忘记将左儿子标记为 null,否则你可能会得到空间溢出或是时间溢出。
1
\
1 2
/ \ \
2 5 => 3
/ \ \ \
3 4 6 4
\
5
\
6
这一题简单粗暴的方法就是用一个list保存所有结点:
class TreeNode: def __init__(self, val): self.val = val self.left, self.right = None, None class Solution: """ @param root: a TreeNode, the root of the binary tree @return: nothing """ def flatten(self, root): list=[] def traverse(root): if(root is None):return list.append(root) traverse(root.left) traverse(root.right) traverse(root) if(len(list)>1): for i in list: root.left=None root.right=i root=root.right a=Tree = TreeNode(1) # b=Tree.left = TreeNode(2) # c=Tree.right = TreeNode(3) # d=b.left=TreeNode(4) s = Solution() s.flatten(a) print(a.right.val) # print(s.flatten(a))当然直接可以用一个遍历来做: