蓝桥杯-历届试题-横向打印二叉树（遍历树+模拟）

//传送门：http://lx.lanqiao.cn/problem.page?gpid=T34
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string>
#include<queue>
#include<string.h>
using namespace std;
#define INF 0x3f3f3f3f
//题意：将整棵树从右到左打印
int fir;      //树根的值， 输入数据中没有重复的数字。
struct tree
{
int num;
struct tree *l,*r;
tree(){l=r=NULL;}
};
void insert(tree* &head,int n)        // 建树
{
if(!head){
head = (tree *) new tree;
head->num = n;
}
else if(n>head->num) insert(head->r,n);
else            insert(head->l,n);
}
void display(tree * head)             // 测试用
{
if(head->l)	display(head->l);
cout<<head->num<<endl;
if(head->r)  display(head->r);
}
string itos(int num)                  // 将整数转为字符串
{
string s="";
int cot = 0;
if(!num) return s="0";
/*if(num<0){                      // 题目没有说明没有负数,但是不特判也能过
s+="-";
num*=-1;
}*/
int tmp[10];
while(num){
tmp[cot++]=num%10;
num /= 10;
}
for(int i=cot-1;i>=0;i--)
s+=tmp[i]+'0';
return s;
}
void dfs(tree *head,string s,int n,string s1) //s 保存全部字符串，s1 保存从跟到当前节点左右移动的关系
{
if(head->num == fir) s+=itos(head->num);
else s+="-|-",s+=itos(head->num);
if(head->r) s1+="1",dfs(head->r,s,n+1,s1),s1 = s1.substr(0,s1.length()-1); //先遍历右子树
int len = s.length();
int cot = 0;
for(int i=0;i<len;i++){
if(s[i]=='|') {                      // 如果是 '|' 需要通过 s1判断能否输出
if(s1[cot]!=s1[cot+1]) cout<<"|";
else cout<<".";
cot++;
}else if(cot<n) cout<<".";           // 如果不是当前位置的数，用'.'代替
else cout<<s[i];                    // 输出当前数
}
if(head->r||head->l) cout<<"-|";         // 如果后面有数 输出'-|'
cout<<endl;
if(head->l) s1+="0",dfs(head->l,s,n+1,s1),s1 = s1.substr(0,s1.length()-1);
}
int main()
{
int n;
tree *head = NULL;
while(~scanf("%d",&n)){
insert(head,n);
}
//display(head);
string s = "";
string s1= "";
fir = head->num;
dfs(head,s,0,s1);
return 0;
}