我们采用深度作为优先级,从根节点开始一直到达某个确定的叶子节点,然后再返回根节点到达另一个分支。深度优先搜索策略可以根据根节点、左孩子树和右孩子树的相对顺序被细分为前序遍历、中序遍历和后序遍历。
前序遍历 (preorder traversal) - 中序遍历 (inorder traversal) - 后序遍历 (postorder traversal)
我们按照高度顺序一层一层的访问整棵树,高层次的节点将会比低层次的节点先被访问到。
下图中的节点依照不同的策略遍历,访问的顺序均为 1, 2, 3, 4, 5
。宽度优先搜索划分层次为 [[1], [2, 3], [4, 5]]
。
递归实现的效率一般比对应的非递归实现低。如果在函数中使用了两个递归调用,效率低下的问题就会变得更为严重。
给定一个二叉树,返回它的前序遍历。
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,2,3]
int* preorderTraversal(struct TreeNode* root, int* returnSize)
输入参数 root
为 NULL
时,如果 return NULL
,*returnSize
必须设置为 0
。
从根节点开始,每次迭代弹出当前栈顶元素,并将其孩子节点压入栈中,先压入右孩子再压入左孩子。输出到最终结果的顺序按照 Top -> Bottom 和 Left -> Right,符合前序遍历的顺序。
时间复杂度:访问每个节点恰好一次,时间复杂度为 O(N)
,其中 N
是节点的个数,即树的大小。
空间复杂度:取决于树的结构,最坏情况存储整棵树,空间复杂度是 O(N)
。
右子节点
是否存在,若存在则 push 到栈中。查看栈顶元素左子节点
,若存在,则 push 到栈中。2.
执行,直到栈空。//============================================================================
// Name : preorder traversal
// Author : Yongqiang Cheng
// Version : Feb 22, 2020
// Copyright : Copyright (c) 2019 Yongqiang Cheng
// Description : Hello World in C++, Ansi-style
//============================================================================
/*
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
/*
* Note: The returned array must be malloced, assume caller calls free().
*/
int* preorderTraversal(struct TreeNode* root, int* returnSize)
{
int idx = 0, front = 0;
struct TreeNode* STACK_DATA[256] = { NULL };
struct TreeNode* pnode = NULL;
int *ret = (int*) malloc(256 * sizeof(int));
*returnSize = 0;
if ((NULL == root) || (NULL == returnSize))
{
return NULL;
}
front = -1;
STACK_DATA[++front] = root;
while (front >= 0)
{
pnode = STACK_DATA[front];
front--;
ret[idx] = pnode->val;
idx++;
if (NULL != pnode->right)
{
STACK_DATA[++front] = pnode->right;
}
if (NULL != pnode->left)
{
STACK_DATA[++front] = pnode->left;
}
}
*returnSize = idx;
return ret;
}
前序遍历 (preorder traversal) 先沿左侧通路自顶而下访问沿途节点,再自底而上依次遍历这些节点的右子树。
https://leetcode-cn.com/