题目:
给你二叉树的根节点 root ,返回它节点值的 前序 遍历。
示例 1:
输入:root = [1,null,2,3]
输出:[1,2,3]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
示例 4:
输入:root = [1,2]
输出:[1,2]
示例 5:
输入:root = [1,null,2]
输出:[1,2]
提示:
- 树中节点数目在范围 [0, 100] 内
- -100 <= Node.val <= 100
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/binary-tree-preorder-traversal
题解:
方法一:递归
前序遍历,按照访问根节点–左子树–右子树的顺序遍历,用递归的方式很容易实现。
/**
* 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().
*/
void preorder(struct TreeNode* root,int* result,int* returnSize){
if(root==NULL) return NULL;
result[(*returnSize)]=root->val;
(*returnSize)++;
preorder(root->left,result,returnSize);
preorder(root->right,result,returnSize);
}
int* preorderTraversal(struct TreeNode* root, int* returnSize){
int* result=malloc(sizeof(int)*2000);
*returnSize=0;
preorder(root,result,returnSize);
return result;
}
运行结果:
方法二:迭代
用迭代的方式来实现方法一的递归,用栈来实现,实质上和递归是一样的,只不过递归的时候是隐式地维护了一个栈。
int* preorderTraversal(struct TreeNode* root, int* returnSize){
int* result=malloc(sizeof(int)*2000);
*returnSize=0;
if(root==NULL) return NULL;
struct TreeNode* stack[2000];
struct TreeNode* node=root;
int top=0;
while(node!=NULL||top>0){
while(node!=NULL){
result[(*returnSize)]=node->val;
(*returnSize)++;
stack[top++]=node;
node=node->left;
}
node=stack[--top];
node=node->right;
}
return result;
}
运行结果:
方法三:Morris 遍历
(参考力扣官方题解)
int* preorderTraversal(struct TreeNode* root, int* returnSize){
int* result=malloc(sizeof(int)*2000);
*returnSize=0;
if(root==NULL) return NULL;
struct TreeNode *p1=root,*p2=NULL;
while(p1!=NULL){
p2=p1->left;
if(p2!=NULL){
while(p2->right!=NULL&&p2->right!=p1){
p2=p2->right;
}
if(p2->right==NULL){
result[(*returnSize)]=p1->val;
(*returnSize)++;
p2->right=p1;
p1=p1->left;
continue;
}
else p2->right=NULL;
}
else{
result[(*returnSize)]=p1->val;
(*returnSize)++;
}
p1=p1->right;
}
return result;
}