时间限制: 1 Sec内存限制: 128 MB提交: 558解决: 139
提交状态
题目描述
给一棵二叉树的层序遍历序列和中序遍历序列,求这棵二叉树的先序遍历序列和后序遍历序列,并给出从右往左、从右上往左下、从上往下分别能看到的结点个数。注意,此处均把二叉树的每条边都设置为等长,角度为45度,因此结点可能在视觉上重叠。所谓从右往左看是指,对同一层的结点,右边的结点会挡住左边的结点,这样同一层结点就只能看到最右边的那一个;同样的,从右上往左下看是指,右上角的结点会挡住左下角45度的结点;从上往下看是指,对同一竖直位置来说,只能看到最上方的结点。
例如对下图来说,从右往左能看到3个结点,从右上往左下能看到3个结点,从上往下能看到5个结点
输入
每个输入文件中一组数据。
第一行一个正整数N(1<=N<=30),代表二叉树的结点个数(结点编号为1~N)。
接下来两行,每行N个正整数,分别代表二叉树的层序遍历序列和中序遍历序列。数据保证序列中1~N的每个数出现且只出现一次。
输出
先输出两行,每行N个正整数,分别代表二叉树的先序遍历序列和后序遍历序列。
接下来分三行输出从右往左、从右上往左下、从上往下分别能看到的结点个数。
每行末尾均不允许输出多余的空格。
样例输入
7
1 2 3 4 5 6 7
4 2 5 1 6 3 7
样例输出
1 2 4 5 3 6 7
4 5 2 6 7 3 1
3
3
5
提示
提交状态
这道题提交了整整9次,最后发现是题目看错,题目要求的是从右上往左下看,而我在求从右下往左上看
首先递归,构建树,这个找找规律就好,之后先序后序输出,直接递归输出即可
最后看到的节点个数
1.从右往做看,这个即求树的层数,只要构建树的时候加个小变量代表当前层数即可
2.从右上往左下看,这个即求往右的最大深度,即只有往右的时候,看到的个数才会+1,同样设置个小变量即可
3.从上往下看,这个可以看成树的宽度,设根节点为位置0,往左就是位置-1,往右就是位置+1,构建一遍树下来可以求出最小位置l和最大位置r
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct node {
int data;
node *l, *r;
int tt;
node() { l = r = NULL; }
};
vector<int> a, b;int N;
vector<node> all;
int lev = 0, lev2 = 0, lev3 = 0;//依次代表看到的个数
int l = 0, r = 0;//代表最左位置和最右位置
void change(vector<int> a, int b1, int b2, int v, int m, int n)//v代表层数,m代表往右的层数,n代表位置
{//构建树
if (m > lev2) lev2 = m;
if (n > r) r = n;
if (n<l) l = n;
if (++v > lev) lev = v;
int pos = find(b.begin(), b.end(), a[0]) - b.begin();
int k = 1;
if (pos > b1)
{
vector<int> temp;
all[a[0]].l = &all[a[k++]];
vector<bool> visited(N + 1, false);
for (int t = b1;t < pos;t++)
visited[b[t]] = true;
for (auto x : a)
if (visited[x]) temp.push_back(x);
change(temp, b1, pos - 1, v, m, n - 1);
}
if (pos < b2)
{
vector<int> temp;
all[a[0]].r = &all[a[k]];
vector<bool> visited(N + 1, false);
for (int t = pos + 1;t <= b2;t++)
visited[b[t]] = true;
for (auto x : a)
if (visited[x]) temp.push_back(x);
change(temp, pos + 1, b2, v, m + 1, n + 1);
}
}
int cnt;
void PreOrderTraver(node *root)//先序遍历
{
if (root == NULL) return;
if (cnt++) cout << " " << root->data;
else cout << root->data;
PreOrderTraver(root->l);
PreOrderTraver(root->r);
}
void PostOrderTraver(node *root)//后序遍历
{
if (root == NULL) return;
PostOrderTraver(root->l);
PostOrderTraver(root->r);
if (cnt++) cout << " " << root->data;
else cout << root->data;
}
int main()
{
cin >> N;
a.resize(N);b.resize(N);all.resize(N + 1);
for (int t = 1;t <= N;t++)
all[t].data = t;
for (int t = 0;t < N;t++)
cin >>a[t];
for (int t = 0;t < N;t++)
cin >> b[t];
change(a, 0, N - 1, 0, 1, 0);
node *root;
root = &all[a[0]];
cnt = 0;
PreOrderTraver(root);
cout << endl;
cnt = 0;
PostOrderTraver(root);
cout << endl;
lev3 = r - l + 1;
cout << lev << endl << lev2 << endl << lev3 << endl;
}