2023-06-14:我们从二叉树的根节点 root 开始进行深度优先搜索。
在遍历中的每个节点处,我们输出 D 条短划线(其中 D 是该节点的深度)
然后输出该节点的值。(如果节点的深度为 D,则其直接子节点的深度为 D + 1
根节点的深度为 0
如果节点只有一个子节点,那么保证该子节点为左子节点
给出遍历输出 S,还原树并返回其根节点 root。
输入:“1-2–3–4-5–6–7”。
输出:[1,2,5,3,4,6,7]。
答案2023-06-14:
大体过程如下:
1.根据输入的遍历字符串 S 来构建一个二叉树。
2.定义一个结构体类型 TreeNode,表示二叉树的节点,包括节点值 Val,左子节点 Left,右子节点 Right。
3.定义一个数组 queue,用于存储节点的深度和值。
4.定义两个全局变量 l 和 r,表示队列的左右指针。
5.定义一个函数 recoverFromPreorder,用于根据遍历字符串 S 还原二叉树。
6.遍历字符串 S 的每一个字符:
a.如果该字符不为 ‘-’(即为数字字符),记录下该数字,直到该数字记录完毕。
b.如果该字符为 ‘-’,则表示该数字已经记录完毕,将该数字加入到 queue 数组中,并将 pickLevel 置为 true。
c.如果该字符是 ‘-’ 或者到达字符串末尾,表示该数字已经记录完毕,将 lvel 记录到队列中, pickLevel 置为 false 。
d.如果该字符是 ‘-’,表示深度加 1;否则,将该数字加入到 number 中。
7.处理掉最后一个数字,将其加入到队列 queue 中。
8.定义一个递归函数 f,用于生成节点,并构建二叉树。
9.取出队列的第一个元素 level,它是当前节点的深度。
10.取出队列的第二个元素 val,它是当前节点的值。
11.生成一个 TreeNode 类型的结构体,元素值为 val,左子节点和右子节点置为 nil。
12.如果队列不为空,且队列的下一个元素的值大于当前节点深度 level,则递归进入左子节点,生成左子树。
13.同样,如果队列不为空,且队列的下一个元素的值大于当前节点深度 level,则递归进入右子节点,生成右子树。
14.返回根节点 head。
时间复杂度为 O(n),其中 n 是遍历字符串 S 的长度。需要遍历字符串 S 一次,并将每个节点入队一次,然后根据队列中的节点数构建二叉树,构建二叉树的时间复杂度也是 O(n)。因此,总时间复杂度为 O(n)。
空间复杂度为 O(n),需要一个数组来存储节点的深度和值,并将其入队。由于二叉树不一定是满二叉树,因此最多需要存储 2n 个节点的深度和值信息。因此,总空间复杂度为 O(n)。
package main
import (
"fmt"
"strconv"
)
type TreeNode struct {
Val int
Left *TreeNode
Right *TreeNode
}
const MAXN = 2001
var queue [MAXN]int
var l, r int
func recoverFromPreorder(traversal string) *TreeNode {
l = 0
r = 0
number := 0
level := 0
pickLevel := true
for i := 0; i < len(traversal); i++ {
c := traversal[i]
if c != '-' {
if pickLevel {
queue[r] = level
level = 0
r++
pickLevel = false
}
d, _ := strconv.Atoi(string(c))
number = number*10 + d
} else {
if !pickLevel {
queue[r] = number
number = 0
r++
pickLevel = true
}
level++
}
}
queue[r] = number
return f()
}
func f() *TreeNode {
level := queue[l]
head := &TreeNode{Val: queue[l+1], Left: nil, Right: nil}
l += 2
if l < r && queue[l] > level {
head.Left = f()
}
if l < r && queue[l] > level {
head.Right = f()
}
return head
}
func main() {
traversal := "1-2--3--4-5--6--7"
result := recoverFromPreorder(traversal)
fmt.Printf("%+v\n", result)
}
use std::cell::RefCell;
use std::rc::Rc;
#[derive(Debug, PartialEq, Eq)]
pub struct TreeNode {
pub val: i32,
pub left: Option<Rc<RefCell<TreeNode>>>,
pub right: Option<Rc<RefCell<TreeNode>>>,
}
impl TreeNode {
#[inline]
pub fn new(val: i32) -> Self {
TreeNode {
val,
left: None,
right: None,
}
}
}
fn recover_from_preorder(traversal: String) -> Option<Rc<RefCell<TreeNode>>> {
let mut queue = vec![0; 2001];
let mut l = 0;
let mut r = 0;
let mut number = 0;
let mut level = 0;
let mut pick_level = true;
for c in traversal.chars() {
if c != '-' {
if pick_level {
queue[r] = level;
level = 0;
r += 1;
pick_level = false;
}
number = number * 10 + c.to_digit(10).unwrap() as i32;
} else {
if !pick_level {
queue[r] = number;
number = 0;
r += 1;
pick_level = true;
}
level += 1;
}
}
queue[r] = number;
let queue_len = r + 1;
f(&mut queue, &mut l, queue_len, 0)
}
fn f(queue: &mut [i32], l: &mut usize, r: usize, level: i32) -> Option<Rc<RefCell<TreeNode>>> {
if *l >= r || queue[*l] != level {
None
} else {
*l += 1;
let head = Rc::new(RefCell::new(TreeNode::new(queue[*l])));
*l += 1;
head.borrow_mut().left = f(queue, l, r, level + 1);
head.borrow_mut().right = f(queue, l, r, level + 1);
Some(head)
}
}
fn main() {
let traversal = String::from("1-2--3--4-5--6--7");
let result = recover_from_preorder(traversal);
println!("{:#?}", result);
}
#include
#include
using namespace std;
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode() : val(0), left(nullptr), right(nullptr) {}
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
TreeNode(int x, TreeNode* left, TreeNode* right) : val(x), left(left), right(right) {}
};
TreeNode* f();
const int MAXN = 2001;
int queue[MAXN];
int l, r;
TreeNode* recoverFromPreorder(string traversal) {
l = 0;
r = 0;
int number = 0;
int level = 0;
bool pickLevel = true;
for (int i = 0; i < traversal.size(); i++) {
char c = traversal[i];
if (c != '-') {
if (pickLevel) {
queue[r++] = level;
level = 0;
pickLevel = false;
}
number = number * 10 + c - '0';
}
else {
if (!pickLevel) {
queue[r++] = number;
number = 0;
pickLevel = true;
}
level++;
}
}
queue[r++] = number;
return f();
}
TreeNode* f() {
int level = queue[l++];
TreeNode* head = new TreeNode(queue[l++]);
if (l < r && queue[l] > level) {
head->left = f();
}
if (l < r && queue[l] > level) {
head->right = f();
}
return head;
}
int main() {
string traversal = "1-2--3--4-5--6--7";
TreeNode* result = recoverFromPreorder(traversal);
cout << result->val << endl;
cout << result->left->val << endl;
cout << result->right->val << endl;
cout << result->left->left->val << endl;
cout << result->left->right->val << endl;
cout << result->right->left->val << endl;
cout << result->right->right->val << endl;
return 0;
}
#include
#include
#include
struct TreeNode {
int val;
struct TreeNode* left;
struct TreeNode* right;
};
struct TreeNode* f();
#define MAXN 2001
int queue[MAXN];
int l, r;
struct TreeNode* recoverFromPreorder(char* traversal) {
l = 0;
r = 0;
int number = 0;
int level = 0;
int pickLevel = 1;
int len = strlen(traversal);
for (int i = 0; i < len; i++) {
char c = traversal[i];
if (c != '-') {
if (pickLevel) {
queue[r++] = level;
level = 0;
pickLevel = 0;
}
number = number * 10 + c - '0';
}
else {
if (!pickLevel) {
queue[r++] = number;
number = 0;
pickLevel = 1;
}
level++;
}
}
queue[r++] = number;
return f();
}
struct TreeNode* f() {
int level = queue[l++];
struct TreeNode* head = (struct TreeNode*)malloc(sizeof(struct TreeNode));
head->val = queue[l++];
head->left = NULL;
head->right = NULL;
if (l < r && queue[l] > level) {
head->left = f();
}
if (l < r && queue[l] > level) {
head->right = f();
}
return head;
}
int main() {
char traversal[] = "1-2--3--4-5--6--7";
struct TreeNode* result = recoverFromPreorder(traversal);
printf("%d\n", result->val);
printf("%d\n", result->left->val);
printf("%d\n", result->right->val);
printf("%d\n", result->left->left->val);
printf("%d\n", result->left->right->val);
printf("%d\n", result->right->left->val);
printf("%d\n", result->right->right->val);
return 0;
}