【题目链接】
【题目考点】
1. 二叉树
【解题思路】
该题意为:为每个结点都设一个长度:叶结点的长度为1,一个非叶结点的长度等于它的左右子树的长度之和。
接着按照先序遍历的顺序,每行输出一个结点的值,这个结点的长度是几,这一行就输出几个字符。
先已知先序遍历序列和中序遍历序列生成整棵树,具体做法见:ybt 1339:【例3-4】求后序遍历
在结点中设成员变量length,表示这个结点的长度。
用calcLen函数递归求出每个结点的长度
- 递归出口:如果该结点是叶子结点,那么长度为1。
- 递归关系:如果该结点不是叶子结点,那么长度为左孩子的长度和右孩子长度的加和。
用preOrder函数递归输出每个结点,每个结点输出的次数是其长度。
【题解代码】
解法1:使用字符数组
#include<bits/stdc++.h>
using namespace std;
#define N 105
struct Node
{
char val;
int left, right;
int length;//表示该结点的长度,应该输出几个字符
};
Node node[N];
int p;
char pre_s[N], mid_s[N];//先序,中序遍历字符串
//由先序遍历序列pre_s[pl]~pre_s[pr]与中序遍历序列mid_s[ml]~mid_s[mr]构建二叉树,返回根结点地址
int createTree(int pl, int pr, int ml, int mr)
{
if(pl > pr || ml > mr)
return 0;
int np = ++p, i;
node[np].val = pre_s[pl];//pre_s[pl]一定是根结点的值
for(i = ml; i <= mr; ++i)
if(mid_s[i] == pre_s[pl])//找到根结点在中序序列中的下标为i
break;
node[np].left = createTree(pl + 1, pl + i - ml, ml, i - 1);
node[np].right = createTree(pl + i - ml + 1, pr, i + 1, mr);
return np;
}
int calcLen(int r)
{
if(r == 0)
return 0;
else if(node[r].left == 0 && node[r].right == 0)//如果地址为r的结点是叶子结点
return node[r].length = 1;//把以r为地址的结点的长度设为1,返回1
else
return node[r].length = calcLen(node[r].left) + calcLen(node[r].right);//把以r为地址的结点的长度为左右孩子结点的长度加和
}
void preOrder(int r)//先序遍历输出整个树的各个结点,每个结点输出次数等于其长度
{
if(r == 0)
return;
for(int i = 0; i < node[r].length; ++i)
cout << node[r].val;
cout << endl;
preOrder(node[r].left);
preOrder(node[r].right);
}
int main()
{
cin >> pre_s >> mid_s;
int len = strlen(pre_s);
int root = createTree(0, len - 1, 0, len - 1);
calcLen(root);
preOrder(root);
return 0;
}
解法2:使用string类
#include<bits/stdc++.h>
using namespace std;
#define N 105
struct Node
{
char val;
int left, right;
int length;//表示该结点的长度,应该输出几个字符
};
Node node[N];
int p;
string s_pre, s_in;
int createTree(string sp, string si)//用先序遍历序列sp与中序遍历序列si构建二叉树,返回树根
{
int np = ++p, i;
node[np].val = sp[0];
for(i = 0; i < si.length(); ++i)
if(sp[0] == si[i])
break;
int len_l = i, len_r = si.length() - 1 - i;//左右子树序列长度
if(len_l > 0)//序列长度大于0,才可以建立一棵树
node[np].left = createTree(sp.substr(1, len_l), si.substr(0, len_l));
if(len_r > 0)
node[np].right = createTree(sp.substr(i+1, len_r), si.substr(i+1, len_r));
return np;
}
int calcLen(int r)
{
if(r == 0)
return 0;
else if(node[r].left == 0 && node[r].right == 0)//如果地址为r的结点是叶子结点
return node[r].length = 1;//把以r为地址的结点的长度设为1,返回1
else
return node[r].length = calcLen(node[r].left) + calcLen(node[r].right);//把以r为地址的结点的长度为左右孩子结点的长度加和
}
void preOrder(int r)//先序遍历输出整个树的各个结点,每个结点输出次数等于其长度
{
if(r == 0)
return;
for(int i = 0; i < node[r].length; ++i)
cout << node[r].val;
cout << endl;
preOrder(node[r].left);
preOrder(node[r].right);
}
int main()
{
cin >> s_pre >> s_in;
int root = createTree(s_pre, s_in);
calcLen(root);
preOrder(root);
return 0;
}