二叉树的存储和遍历

最新推荐文章于 2024-07-12 18:56:03 发布

gaomuchen_120309

最新推荐文章于 2024-07-12 18:56:03 发布

阅读量90

点赞数

文章标签：算法深度优先图论

本文链接：https://blog.csdn.net/gaomuchen_120309/article/details/131348916

版权

请添加图片描述

题目：

二叉树的遍历

时间限制： $1$ 秒内存限制： $128 M$

题目描述

给定一棵二叉树，根据中序遍历的顺序，找到其中值为x的点（可能不止一个），说出它的中序遍历的上一个点是谁（如果是第一个点，则输出 $- 1$ ）。

输入描述

第一行两个数 $n$ 和 $x$ ，表示有 $n$ 个点，并且要找的值为 $x （ 0 < n ， x <= 100)$ 。
接下来一行有 $n$ 个数 $a_i$ ，第 $i$ 个数表示第 $i$ 个节点的数值为 $a_i$ 。

输出描述

输出每个 $x$ 中序遍历的前一个数

样例

输入

5 3
1 3 2 5 3

输出

5
3

提示

树的中序遍历序列为 $[5, 3, 3, 1, 2]$
第一个 $3$ 前面的数为 $5$ ，第二个3前面的数为 $3$

思路及代码

解题思路

输入 $n, x$ 和二叉树。

cin>>n>>x;
for(int i=1; i<=n; i++) {
	cin>>a[i];
}

中序遍历二叉树，存储中序遍历序列。

dfs(1);

void dfs(int k) {
	if(k*2<=n && a[k*2]!=0) {
		dfs(k*2);
	}
	zhong[++cnt]=a[k];
	if(k*2+1<=n && a[k*2+1]!=0) {
		dfs(k*2+1);
	}
}

遍历中序遍历序列，查找 $x$ ，找到输出它前面那个数，如果它是第一个，输出 $- 1$ 。

for(int i=1; i<=n; i++) {
	if(zhong[i]==x) {
		if(i-1>0) {
			cout<<zhong[i-1]<<endl;
		} 
		else {
			cout<<-1<<endl;
		}
	}
}

AC代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int a[102], zhong[102], cnt=0;
int n, x;
void dfs(int k) {
	if(k*2<=n && a[k*2]!=0) {
		dfs(k*2);
	}
	zhong[++cnt]=a[k];
	if(k*2+1<=n && a[k*2+1]!=0) {
		dfs(k*2+1);
	}
}
int main() {
	cin>>n>>x;
	for(int i=1; i<=n; i++) {
		cin>>a[i];
	}
	dfs(1);
	for(int i=1; i<=n; i++) {
		if(zhong[i]==x) {
			if(i-1>0) {
				cout<<zhong[i-1]<<endl;
			} 
			else {
				cout<<-1<<endl;
			}
		}
	}
	return 0;
}