#include
#include
using namespace std;
struct MyTreeNode
{
int col;//打印結果中的列
int row;//打印結果中的行
char data;
MyTreeNode rightChild , *leftChild;
};
MyTreeNode rebuild(string preorder,string inorder)//先序,中序
{
int i , len ;
MyTreeNode *root = new MyTreeNode() ;
root->data = preorder[0] ;
len = inorder.length() ;
for(i=0;i
{
if(preorder[0]==inorder[i])
{
if(i!=0)
{ //左子樹
root->leftChild=rebuild(preorder.substr(1,i),inorder.substr(0,i));
}
else
{
root->leftChild = NULL ;
}
if(i!=len-1)
{//右子樹
root->rightChild=rebuild(preorder.substr(i+1,len-1-i),inorder.substr(i+1,len-1-i));
}
else
{
root->rightChild = NULL ;
}
}
}
return root ;
};
void after(MyTreeNode *root)
{
if(root==NULL)return ;
{
if(root->leftChild!=NULL)
after(root->leftChild) ;
if(root->rightChild!=NULL)
after(root->rightChild) ;
cout << root->data;
}
};
const int SPAN=4;//每一層的橫向縮進值為4
unsigned char printResult[9][SPAN*4];//打印結果數組
int currRow=0;//第一輪行計數器
void printArray(unsigned char array[][SPAN*4],int length0,int length1)
{//打印結果數組
for(int i=0;i
{//結果數組的行
for(int j=0;j
{//結果數組的列
unsigned char p=array[i][j];
if(p==0)
{//若字符值為0則換為空格
p=' ';
}
cout
}
cout
}
}
//確定每個節點在結果中的行列
void preorderTraverse(MyTreeNode*root,int level)
{
if(root==NULL)return;
root->row=currRow;
currRow++;
root->col=level*SPAN;
printResult[root->row][root->col]=root->data;
preorderTraverse(root->leftChild,level+1);
preorderTraverse(root->rightChild,level+1);
}
//填充父節點與子節點的連線
void preorderForLine(MyTreeNode*root)
{
if(root==NULL)return;
int sCol=root->col;
int sRow=root->row;
if(root->leftChild!=NULL)
{//當前子樹根節點到左子節點的連線
int eCol=root->leftChild->col;
int eRow=root->leftChild->row;
for(int i=sRow+1;i<=eRow;i++)
{//豎線
printResult[i][sCol]=179;
}
for(int i=sCol+1;i
{ //橫線
printResult[eRow][i]=196;
}
}
if(root->rightChild!=NULL)
{//當前子樹根節點到右子節點的連線
int eCol=root->rightChild->col;
int eRow=root->rightChild->row;
for(int i=sRow+1;i<eRow;i++)
{//豎線
printResult[i][sCol]=179;
}
for(int i=sCol+1;i<eCol;i++)
{//橫線
printResult[eRow][i]=196;
}
}
preorderForLine(root->leftChild);
preorderForLine(root->rightChild);
}
//掃描結果數組將其右側有橫線的豎線替換為豎橫線
void processAfterOne(unsigned char array[][SPAN*4],int length0,int length1)
{
for(int i=0;i
{//結果數組的行
for(int j=0;j
{//結果數組的列
unsigned char p=array[i][j];
if (p==179)//若字符值為179豎線
{//若字符值為179豎線則查看其右側字符是否為橫線
unsigned char pr=array[i][j+1];
if(pr==196)
{//若其右側為橫線則將其替換為豎橫線
array[i][j]=195;
}
}
}
}
}
//掃描結果數組將其中下面沒有豎橫線替換為右拐豎線
void processAfterTwo(unsigned char array[][SPAN*4],int length0,int length1)
{
for(int i=0;i
{// 結果數組的行
for(int j=0;j
{//結果數組的列
unsigned char p=array[i][j];
if(p==195) //若字符值為195豎線
{//若字符值為195豎橫線則查看其下側字符是否為空白
if(i+1
{
unsigned char pr=array[i+1][j];
if(pr=0)
{//若其右側為橫線則將其替換為豎橫線
array[i][j]=192;
}
}
else
{//若其位於最後一行則將其替代為右拐豎線
array[i][j]=192;
}
}
}
}
}
int main(int argc, char *argv[])
{
system("chcp 437>nul.");
string preorder , inorder ;
MyTreeNode *root = NULL ;
cout<<"please input preorder:"<
cin>>preorder;
cout<<"please input inorder:"<
cin>>inorder;
cout<<"post:"<<endl;
root = rebuild(preorder,inorder) ;
after(root) ;
cout<<endl;
cout<<"=====preorder traverse print binary tree====="<<endl;
cout<<"================"<<endl;
//先序遍歷(計算每個節點打印位置)
preorderTraverse(root,0);
printArray(printResult,preorder.length(),SPAN*4);
cout<<"================="<<endl;
//先序遍歷(填充父子節點到子節點的連線)
preorderForLine(root);
printArray(printResult,preorder.length(),SPAN*4);
cout<<"===================="<<endl;
//掃描結果數組將其中右側有橫線的豎線替換為豎橫線
processAfterOne(printResult,preorder.length(),SPAN*4);
printArray(printResult,preorder.length(),SPAN*4);
cout<<"===================="<<endl;
//掃描結果數組將其中的下面沒有豎線的豎橫線替換為右拐豎線
processAfterTwo(printResult,preorder.length(),SPAN*4);
printArray(printResult,preorder.length(),SPAN*4);
cout<<"===================="<<endl;
return 0;
}
substr(n,i)
從字符串中第n個字符開始,取i個,構成一個子串