一本通 1367:查找二叉树(tree_a)

1367:查找二叉树(tree_a)


时间限制: 1000 ms         内存限制: 65536 KB
提交数: 3708     通过数: 2278

【题目描述】

已知一棵二叉树用邻接表结构存储,中序查找二叉树中值为x的结点,并指出是第几个结点。例:如图二叉树的数据文件的数据格式如下:

【输入】

第一行n为二叉树的结点个树,n<=100;第二行x表示要查找的结点的值;以下第一列数据是各结点的值,第二列数据是左儿子结点编号,第三列数据是右儿子结点编号。

【输出】

一个数即查找的结点编号。

【输入样例】

7
15
5 2 3
12 4 5
10 0 0
29 0 0
15 6 7
8 0 0
23 0 0
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
优化下面代码class TreeNode: def __init__(self, val): self.val = val self.left = None self.right = None root = TreeNode('a') root.left = TreeNode('b') root.right = TreeNode('c') root.left.left = TreeNode('d') root.left.right = TreeNode('e') root.right.left = TreeNode('f') root.right.right = TreeNode('g') root.left.left.left = TreeNode('h') root.left.left.right = TreeNode('i') def preorder_traversal(root): if not root: return print(root.val, end=' ') preorder_traversal(root.left) preorder_traversal(root.right) def inorder_traversal(root): if not root: return inorder_traversal(root.left) print(root.val, end=' ') inorder_traversal(root.right) def postorder_traversal(root): if not root: return postorder_traversal(root.left) postorder_traversal(root.right) print(root.val, end=' ') from collections import deque def level_order_traversal(root): if not root: return queue = deque() queue.append(root) while queue: node = queue.popleft() print(node.val, end=' ') if node.left: queue.append(node.left) if node.right: queue.append(node.right) def get_height(root): if not root: return 0 left_height = get_height(root.left) right_height = get_height(root.right) return max(left_height, right_height) + 1 def get_node_count(root): if not root: return 0 left_node_count = get_node_count(root.left) right_node_count = get_node_count(root.right) return left_node_count + right_node_count + 1 print("先序遍历:") preorder_traversal(root) print("中序遍历:") inorder_traversal(root) print("后序遍历:") postorder_traversal(root) print("层次遍历:") level_order_traversal(root) print("该二叉树的高度为:") get_height(root) print("该二叉树的节点个数为 ") get_node_count(root)
06-01
这段代码实现了二叉树的四种遍历方式和获取二叉树的高度和节点个数。代码本身没有什么问题,但是可以考虑进行一些优化。 1. 可以在遍历时使用yield关键字,将遍历结果逐个返回,而不是一次性打印出来。这样可以节省内存空间,避免一次性将所有结果存储在内存中。 2. 可以使用迭代方式代替递归方式实现遍历,这样可以避免递归深度过大导致的栈溢出问题。 3. 可以将获取二叉树高度和节点个数的函数合并成一个函数,减少重复代码。 下面是优化后的代码示例: ```python class TreeNode: def __init__(self, val): self.val = val self.left = None self.right = None def preorder_traversal(root): if not root: return yield root.val yield from preorder_traversal(root.left) yield from preorder_traversal(root.right) def inorder_traversal(root): if not root: return yield from inorder_traversal(root.left) yield root.val yield from inorder_traversal(root.right) def postorder_traversal(root): if not root: return yield from postorder_traversal(root.left) yield from postorder_traversal(root.right) yield root.val def level_order_traversal(root): if not root: return queue = [root] while queue: node = queue.pop(0) yield node.val if node.left: queue.append(node.left) if node.right: queue.append(node.right) def get_tree_info(root): if not root: return 0, 0 left_height, left_node_count = get_tree_info(root.left) right_height, right_node_count = get_tree_info(root.right) return max(left_height, right_height) + 1, left_node_count + right_node_count + 1 root = TreeNode('a') root.left = TreeNode('b') root.right = TreeNode('c') root.left.left = TreeNode('d') root.left.right = TreeNode('e') root.right.left = TreeNode('f') root.right.right = TreeNode('g') root.left.left.left = TreeNode('h') root.left.left.right = TreeNode('i') print("先序遍历:", list(preorder_traversal(root))) print("中序遍历:", list(inorder_traversal(root))) print("后序遍历:", list(postorder_traversal(root))) print("层次遍历:", list(level_order_traversal(root))) height, node_count = get_tree_info(root) print("该二叉树的高度为:", height) print("该二叉树的节点个数为:", node_count) ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值