[LeetCode] Binary Tree Preorder Traversal

简介: This is a fundamental and yet classic problem. I share my three solutions here: Iterative solution using stack --- O(n) time and O(n) space; Recurs...

This is a fundamental and yet classic problem. I share my three solutions here:

  1. Iterative solution using stack --- O(n) time and O(n) space;
  2. Recursive solution --- O(n) time and O(n) space (considering the spaces of call stack);
  3. Morris traversal --- O(n) time and O(1) space!!!

Iterative solution using stack:

 1     vector<int> preorderTraversal(TreeNode* root) {
 2          vector<int> nodes;
 3          stack<TreeNode*> toVisit;
 4          TreeNode* curNode = root;
 5          while (curNode || !toVisit.empty()) {
 6              if (curNode) {
 7                  nodes.push_back(curNode -> val);
 8                  if (curNode -> right) toVisit.push(curNode -> right);
 9                  curNode = curNode -> left;
10              }
11              else {
12                  curNode = toVisit.top();
13                  toVisit.pop();
14              }
15          }
16          return nodes;
17     }

Recursive soltuion:

 1 void preorder(TreeNode* root, vector<int>& nodes) {
 2     if (!root) return;
 3     nodes.push_back(root -> val);
 4     preorder(root -> left, nodes);
 5     preorder(root -> right, nodes);
 6 } 
 7 vector<int> preorderTraversal(TreeNode* root) {
 8     vector<int> nodes;
 9     preorder(root, nodes);
10     return nodes;
11 } 

Morris traversal:

 1 vector<int> preorderTraversal(TreeNode* root) {
 2     TreeNode* curNode = root;
 3     vector<int> nodes;
 4     while (curNode) {
 5         if (curNode -> left) {
 6             TreeNode* predecessor = curNode -> left;
 7             while (predecessor -> right && predecessor -> right != curNode)
 8                 predecessor = predecessor -> right;
 9             if (!(predecessor -> right)) {
10                 nodes.push_back(curNode -> val);
11                 predecessor -> right = curNode;
12                 curNode = curNode -> left;
13             }
14             else {
15                 predecessor -> right = NULL;
16                 curNode = curNode -> right;
17             }
18         }
19         else {
20             nodes.push_back(curNode -> val);
21             curNode = curNode -> right;
22         }
23     }
24     return nodes;
25 }

 

目录
相关文章
|
Java
Leetcode 114. Flatten Binary Tree to Linked List
思路也很简单,先把root的左子树(如有)变成单链表 leftlinkedlist,把root的右子树(如有)变成单链表 rightlinkedlist,再把root的右节点变成leftlikedlist,再把rightlinkedlist接到leftlinkedlist后面,代码如下。
42 1
Leetcode Minimum Depth of Binary Tree (面试题推荐)
计算树的最小深度 很简单的一道题,只需要遍历一次树,到叶子节点的时候计算一下深度和当前最小深度比较,保存最小值就行。 我在这用了一个全局变量 mindepth。总感觉我这代码写的不够简练,求更精简的方法。
47 0
Leetcode Binary Tree Postorder Traversal(面试题推荐)
非递后续归遍历二叉树,肯定得用到栈。先序遍历很好写,但后续遍历就不是那么容易了。 只需要设置个指针pre,指向最后输出的那个节点就行了,只要判断cur指针指向的是上次输出节点的父节点,且cur无其他未遍历的节点,这个时候就把cur节点输出即可,然后更改pre。原理是要遍历当前节点,其所有子节点都必须遍历完,因为肯定是先左后右,所以只需一个指针保持前一次输出的结果即可。
50 0
Leetcode 236. Lowest Common Ancestor of a Binary Tree
根据LCA的定义,二叉树中最小公共祖先就是两个节点p和q最近的共同祖先节点,LCA的定义没什么好解释的,主要是这道题的解法。
36 0
Leetcode 623. Add One Row to Tree
题目很简单,在树的第d层加一层,值为v。递归增加一层就好了。代码如下
46 0
LeetCode contest 190 5418. 二叉树中的伪回文路径 Pseudo-Palindromic Paths in a Binary Tree
LeetCode contest 190 5418. 二叉树中的伪回文路径 Pseudo-Palindromic Paths in a Binary Tree
|
算法 Python
LeetCode 108. 将有序数组转换为二叉搜索树 Convert Sorted Array to Binary Search Tree
LeetCode 108. 将有序数组转换为二叉搜索树 Convert Sorted Array to Binary Search Tree
LeetCode 102. 二叉树的层序遍历 Binary Tree Level Order Traversal
LeetCode 102. 二叉树的层序遍历 Binary Tree Level Order Traversal
|
2月前
|
Unix Shell Linux
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
本文提供了几个Linux shell脚本编程问题的解决方案,包括转置文件内容、统计词频、验证有效电话号码和提取文件的第十行,每个问题都给出了至少一种实现方法。
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
|
3月前
|
搜索推荐 索引 Python
【Leetcode刷题Python】牛客. 数组中未出现的最小正整数
本文介绍了牛客网题目"数组中未出现的最小正整数"的解法,提供了一种满足O(n)时间复杂度和O(1)空间复杂度要求的原地排序算法,并给出了Python实现代码。
108 2