1368:对称二叉树(tree_c)

【算法分析】
        题目给的输入的序列是二叉树的顺序存储结构,设数组tree,tree[i]表示用顺序存储结构保存二叉树第i结点的值。那么输入的第i个字符就是tree[i]的值。这里可以使用每次循环读入一个字符的方法,当读入EOF时停止读入。也可以先读入整个字符串,再做赋值。为tree数组设值。设标志位isSym表示该树是否是对称的。
        遍历整棵树(使用哪种遍历方法都可以),看每个结点的左右孩子,如果左右孩子中有1个是空结点而另一个不是,那么整棵树就不是对称的。遍历完所有结点后,可以确定该树是否是对称的。
       注意在遍历顺序存储结构的树的时候,临时求出的下标可能会很大(因为每次都会乘以2),所以数组尽量开得大一些(比如最后一个结点的地址是1000,那么要把数组长度开成2000多),也可以先记录整个顺序存储结构tree数组中最后一个元素的下标tn,如果访问到的下标超过tn,那么这次访问无意义,直接返回。

【参考代码】

#include <bits/stdc++.h>
using namespace std;
#define N 1005
char tree[N];//树的顺序存储结构,初值都为'\0',tree[i]:第i结点的值,如果为'\0'表示这里为空。
//tree[2*i]:第i结点的左孩子的值,tree[2*i+1]:第i结点右孩子的值
int tn;//tree数组最后一个元素的下标,应该保证N > 2*tn 
bool isSymmetric(int r)//判断以r为根结点的树是否是对称的 
{
	if(tree[r] == '\0')//如果遍历到空结点,那么是对称的 
		return true;
	if(tree[2*r] == '\0' && tree[2*r+1] != '\0' || 
		tree[2*r] != '\0' && tree[2*r+1] == '\0')//如果左子树空但右子树不空,或右子树空但左子树不空 
		return false;
	return isSymmetric(2*r) && isSymmetric(2*r+1);//左右子树都是对称的,该树才是对称的 
}
int main()
{
	char c;
	while((c = cin.get()) != EOF)//二叉树的顺序存储结构字符串读入tree 
	{
		if(c == '#' || c == '\n')//遇到'#'或换行符,都转为0 
			tree[++tn] = '\0';
		else
			tree[++tn] = c;
	}
	cout << (isSymmetric(1) ? "Yes" : "No");
	return 0;
} 

  • 3
    点赞
  • 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、付费专栏及课程。

余额充值