蓝桥杯 历届试题 PREV-11 横向打印二叉树 排序二叉树

历届试题 横向打印二叉树
时间限制：1.0s 内存限制：256.0MB
提交此题 锦囊1 锦囊2
问题描述
二叉树可以用于排序。其原理很简单：对于一个排序二叉树添加新节点时，先与根节点比较，若小则交给左子树继续处理，否则交给右子树。
当遇到空子树时，则把该节点放入那个位置。
比如，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

分析：本题就是打印一个排序二叉树，左根右按照由小到大排序。按题意来说，就是横向打印，由上到下的数值依次减小。
　　排序二叉树相信大家都不难，唯一烦人的地方在于怎么按照要求去打印，这时我们就要发现一些规律。
　　从图中和题意来看，不难理解，打印出来的图案是一个横向的排序二叉树，而对应的字符’-‘和’|’，则代表树的边。具体规则如下：
　　1. 如果该结点有子树，则数字的后面有’-’；
　　2. 如果该结点有双亲结点，则数字的前面有’-’，不过除了根结点以外，其他结点都有双亲结点，所以就不用设双亲结点指针来判断了；
　　3. 如果该树有左孩子，则字符’-‘后面开始，一直向下到左孩子所在的行，都有’|’；
　　4. 如果该树有右孩子，则字符’-‘后面开始，一直向上到右孩子所在的行，都有’|’；
　　5. 当树画完之后，其余空闲位置，都是字符’.’；
　　具体代码如下：

PS：这道题我当时删删改改很多次，懒得整理了，算是我发的代码中较乱的了。不过注释依然是有的，但如果有哪没看懂的，欢迎随时留言。
主要的操作就是printdp这个子函数，将排序二叉树经过处理写入二维数组中，这步操作大家可以借鉴一下。当然，如果有更好的方法，也欢迎随时留言，大家互相探讨。

#include <iostream>
#include <vector>
#include <cstring>
using namespace std;

//二叉树的结构体 
struct tree
{
	int data;								//当前节点的数据 
	int len;								//当前节点数字的长度 
	int c, y;								//c:当前节点的在树中的层数, y:横向打印的纵坐标 
	tree *left, *right;						//左、右子树 
	tree() { left = right = NULL;}	//树的初始化 
};

vector<int>		a;					//存输入的数据 
tree			*head = new tree;	//排序二叉树 
char			**dp;				//创建二维数组用来打印输出的图形
int				*top;				//创建数组用来存打印时每行的长度
int				cd = 0;				//判断当前节点大小的顺序 

//计算数字的长度 
int intlen(int num)
{
	int res = 1;
	while(num /= 10)
	{
		res++;
	}
	return res;
}

//右根左写入y 
void printy(tree *h)
{
	if(h)
	{
		printy(h->right);
		h->y = cd++;
		printy(h->left);
	}
}

//写入排序二叉树
void dfs(tree *&h, int x, int c)
{
	if(!h)
	{	//如果当前结点为空，则创建并写入数据
		h = new tree;
		h->data = x;
		h->len = intlen(x);
		h->c = c;
		return;
	}
	
	if(x < h->data)	dfs(h->left, x, c+1);	//小的放在左边 
	else			dfs(h->right, x, c+1);	//大的放在右边 
}

//填写树的图案，深度遍历
void printdp(tree *h, int x)
{
	if(h)
	{
		//写前面的'-' 
		if(h->c != 1) dp[h->y][x++] = '-';
		
		//写当前节点的值 
		sprintf(&dp[h->y][x], "%d", h->data);
		x += h->len;
		
		//写后面的'-'
		if(h->left || h->right)
			dp[h->y][x++] = '-';
		
		//写入连接子树的竖线'|' 
		if(h->right)
		{	//画连接右子树的竖线'|' 
			for(int i = h->right->y; i <= h->y; i++)
			{
				dp[i][x] = '|';
			}
		}
		if(h->left)
		{	//画连接左子树的竖线'|' 
			for(int i = h->y; i <= h->left->y; i++)
			{
				dp[i][x] = '|';
			}
		}
		
		top[h->y] = x;	//存该行最后一个元素的下标
		
		printdp(h->right, x+1);
		printdp(h->left, x+1);
	}
}

int main()
{
	int n = 0;
	
	//输入
	for(int i = 0; cin >> i; n++)
		a.push_back(i);
	top = new int[n];
	dp = new char*[n];
	for(int i = 0; i < n; i++)
	{
		dp[i] = new char[n*5];
		for(int j = 0; j < 5 * n; j++)
			dp[i][j] = 0;
	}
	
	//存储 
	//存根节点
	head->data = a[0];
	head->len = intlen(a[0]);
	head->c = 1;
	//存其他结点
	for(vector<int>::iterator it = a.begin() + 1; it != a.end(); it++)
		dfs(head, *it, 1);
	//写入结点的参数y
	printy(head);
	//写入树的图案
	printdp(head, 0);
	//填充空白位置为'.'
	for(int i = 0; i < n; i++)
		for(int j = 0; j < top[i]; j++)	//每行不超过top[i]
			if(!dp[i][j]) dp[i][j] = '.';
	
	//输出图案结果
	for(int i = 0; i < n; i++)
	{
		cout << dp[i] << endl;
	}
	
	return 0;
}