蓝桥杯横向打印二叉树代码解析:PT(int root,int L) 函数详解
这段代码实现了二叉树的横向打印。具体来说:
- 函数传入参数为当前节点的编号 (
root) 和当前节点的深度 (L)(即该节点所在的行数)。 - 如果当前节点有右子树,则先递归打印右子树,同时将当前节点的深度加上该节点的数据长度(即该节点的值所占的字符数)和一些特定字符的长度(如果该节点是根节点,则需要加上 2 个字符长度;如果不是根节点,则需要加上 4 个字符长度)。
- 接着在字符串数组中为该节点所在的行添加一些空格字符,以保证该节点在该行的位置正确。
- 记录该节点在字符串数组中的位置。
- 将该节点的数据值和一些特定字符(如果该节点有左子树或右子树,则需要添加 '|-’ 和 '-|’ 字符)添加到该节点所在行的字符串数组中。
- 如果该节点有左子树或右子树,则记录该节点在该行的位置(即当前节点的深度加上该节点的数据长度和一些特定字符的长度)。
- 最后如果该节点有左子树,则递归打印左子树,同时将当前节点的深度加上该节点的数据长度和一些特定字符的长度。
代码示例:
void PT(int root,int L)
{
if(tree[root].rson!=-1) PT(tree[root].rson,L+len(tree[root].data)+(root==0?2:4)-1);
for(int i=0;i<L;i++) str[cnt][i]='.';
tree[root].v=cnt;
sprintf(str[cnt++]+L,'%s%d%s',(root==0?'':'|-'),tree[root].data,((tree[root].lson==-1&&tree[root].rson==-1)?'':'-|'));
if(tree[root].lson!=-1 || tree[root].rson!=-1) tree[root].p=L+len(tree[root].data)+(root==0?2:4)-1;
if(tree[root].lson!=-1) PT(tree[root].lson,L+len(tree[root].data)+(root==0?2:4)-1);
}
代码解析:
tree:表示二叉树的结构,tree[root]表示根节点的结构,tree[root].data表示根节点的值,tree[root].lson和tree[root].rson分别表示根节点的左子树和右子树的编号。str:表示用于存储打印结果的字符串数组。cnt:表示当前字符串数组的索引,用于记录当前节点在字符串数组中的位置。len(tree[root].data):表示当前节点的值所占的字符数。root==0?2:4:判断当前节点是否是根节点,如果是则加 2 个字符长度,否则加 4 个字符长度。'|':表示垂直方向的连接线。'-':表示水平方向的连接线。
代码逻辑:
- 递归打印右子树: 如果当前节点有右子树,则先递归打印右子树,同时将当前节点的深度加上该节点的数据长度和一些特定字符的长度,并将新的深度作为参数传递给
PT函数。 - 添加空格: 在字符串数组中为该节点所在的行添加一些空格字符,以保证该节点在该行的位置正确。
- 记录位置: 记录该节点在字符串数组中的位置,并将其存放在
tree[root].v中。 - 添加节点值: 将该节点的数据值和一些特定字符添加到该节点所在行的字符串数组中。
- 记录连接点: 如果该节点有左子树或右子树,则记录该节点在该行的位置(即当前节点的深度加上该节点的数据长度和一些特定字符的长度),并将该位置存放在
tree[root].p中。 - 递归打印左子树: 如果该节点有左子树,则递归打印左子树,同时将当前节点的深度加上该节点的数据长度和一些特定字符的长度,并将新的深度作为参数传递给
PT函数。
总结: 该代码利用递归算法,通过遍历二叉树的每个节点,并在每个节点处添加适当的空格和特定字符,最终实现了二叉树的横向打印。
原文地址: https://www.cveoy.top/t/topic/mqUh 著作权归作者所有。请勿转载和采集!