蓝桥杯-横向打印二叉树（模拟）

横向打印二叉树：http://lx.lanqiao.org/problem.page?gpid=T34

问题描述

二叉树可以用于排序。其原理很简单：对于一个排序二叉树添加新节点时，先与根节点比较，若小则交给左子树继续处理，否则交给右子树。

当遇到空子树时，则把该节点放入那个位置。

比如，10 8 5 7 12 4 的输入顺序，应该建成二叉树如下图所示，其中.表示空白。

...|-12

10-|

...|-8-|

.......|...|-7

.......|-5-|

...........|-4

本题目要求：根据已知的数字，建立排序二叉树，并在标准输出中横向打印该二叉树。

输入格式

输入数据为一行空格分开的N个整数。 N<100，每个数字不超过10000。

输入数据中没有重复的数字。

输出格式

输出该排序二叉树的横向表示。为了便于评卷程序比对空格的数目，请把空格用句点代替：

样例输入1

10 5 20

样例输出1

...|-20

10-|

...|-5

样例输入2

5 10 20 8 4 7

样例输出2

.......|-20

..|-10-|

..|....|-8-|

..|........|-7

5-|

..|-4

这题好麻烦，目前还没想到可以不保存在数组中直接输出的方法

感觉题目的样例不能准确表明各种格式

大概：

465 32413 1231 45 17 674 14 5756

这组样例（不要在意范围）应该输出

....|-32413-|

....|.......|......|5756

....|.......|-1231-|

....|..............|-674

465-|

....|-45-|

.........|-17-|

..............|-14

就对了

#include <cstdio>
#include <algorithm>

using namespace std;

struct Node {
int lson,rson,num,r;
Node() {
lson=rson=num=-1;
}
}tr[105];
int a,cnt,l=1,sta;
char ans[105][605];

inline int getLen(int n) {//返回数字n的长度
cnt=n<=0?1:0;
while(n) {
++cnt;
n/=10;
}
return cnt;
}

void add(int i,int lv) {//向排序二叉树添加一个数a
if(tr[i].num>a) {
if(tr[i].lson==-1) {
tr[i].lson=l;
tr[l++].num=a;
return ;
}
add(tr[i].lson,lv+1);
}
else {
if(tr[i].rson==-1) {
tr[i].rson=l;
tr[l++].num=a;
return ;
}
add(tr[i].rson,lv+1);
}
}

void print(int i,int lv,int num) {//将横向二叉树初步输出到ans数组中
if(tr[i].rson!=-1)
print(tr[i].rson,lv+1,num+getLen(tr[i].num)+(lv==1?1:3));

int j=0;
for(;j<num;++j)
ans[sta][j]='.';
sprintf(ans[sta]+j,"%s%d%s",lv==1?"\0":"|-",tr[i].num,((tr[i].lson==-1&&tr[i].rson==-1)?"\n\0":"-|\n\0"));
tr[i].r=sta++;
if(tr[i].lson!=-1)
print(tr[i].lson,lv+1,num+getLen(tr[i].num)+(lv==1?1:3));
}

void format(int i,int lv,int num) {//将初步得到的ans数组进行格式化，即添加'|'
if(tr[i].rson!=-1) {
int nxt=num+getLen(tr[i].num)+(lv==1?1:3);
format(tr[i].rson,lv+1,nxt);
for(int j=tr[tr[i].rson].r;j<tr[i].r;++j)
ans[j][nxt]='|';
}
if(tr[i].lson!=-1) {
int nxt=num+getLen(tr[i].num)+(lv==1?1:3);
format(tr[i].lson,lv+1,nxt);
for(int j=tr[i].r+1;j<=tr[tr[i].lson].r;++j)
ans[j][nxt]='|';
}
}

int main() {
scanf("%d",&a);
tr[0].num=a;
while(1==scanf("%d",&a))
add(0,1);
sta=0;
print(0,1,0);
format(0,1,0);
for(int i=0;i<sta;++i)
printf("%s",ans[i]);
return 0;
}