给定一个二叉树的根节点 root
,返回 它的 中序 遍历 。
示例 1:
输入:root = [1,null,2,3] 输出:[1,3,2]
示例 2:
输入:root = [] 输出:[]
示例 3:
输入:root = [1] 输出:[1]
提示:
- 树中节点数目在范围
[0, 100]
内 -100 <= Node.val <= 100
【方法一】递归算法
二叉树的中遍历按照:左-根-右来进行。
针对主函数,我们需要定义一个数组a,将遍历的树中元放在数组中;returnSize指针所指向的值初始化为0,表示结果数组的大小初始为0;调用inorder函数,最后返回数组a。
针对调用函数,首先进行判断是否为空,为空的话直接进行返回,不需要进行任何操作。然后按照左-根-右来进行遍历。针对根节点处理:a[(*returnSize)++] = root->val。将根节点保持到数组中,然后将returnSize+1,进行下一个节点操作。
具体案例分析:
root = [1,null,2,3],输出:[1,3,2]。
针对主函数,开始时a[0]={},returnSize=0,然后调用inorder(root, a, returnSize)。
由于 root 不为空,继续递归调用 inorder(root->left, a, returnSize),但 root->left 为空,因此直接返回。
访问 root->val,即1,将其添加到数组 a[0],并将 *returnSize 递增为1。
递归调用 inorder(root->right, a, returnSize)。
同样地,由于 root->right 不为空,继续递归调用 inorder(root->right, a, returnSize)。
进入右子树后,此时的root=2不为空,那么开始遍历左子树 inorder(root->left, a, returnSize),进入也就是根为3的这棵树。
然后继续inorder(root->left, a, returnSize),此时3的左子树为空,直接返回。访问根节点3,将其添加到数组 a[1],并将 *returnSize 递增为2。然后访问右子树inorder(root->right, a, returnSize)。,为空,那么跳出,根为2的这棵树的左子树访问完毕,访问根节点2,将其添加到数组 a[2],并将 *returnSize 递增为3,然后访问右子树,为空,直接返回。
最终,数组 a 包含元素 [1, 3, 2],并且 *returnSize 为3。
返回数组 a。
(整个流程可以描述为,root=1-> inorder(root->left, a, returnSize)为空->返回1->inorder(root->right, a, returnSize)进入右子树,此时root=2-> inorder(root->left, a, returnSize)进入root=2的左子树->inorder(root->left, a, returnSize)root=3-> inorder(root->left, a, returnSize)为空->返回3->inorder(root->right, a, returnSize)为空,此时2的左子树判断结束->返回2->inorder(root->right, a, returnSize)为空,循环和调用结束。返回a[1,3,2])
具体C语言代码如下:
/**
* 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 inorder(struct TreeNode* root, int* a, int* returnSize) {//递归函数执行左-根-右
if (!root) {
return;
}
inorder(root->left, a, returnSize);//左
a[(*returnSize)++] = root->val;//根
inorder(root->right, a, returnSize);//右
}
int* inorderTraversal(struct TreeNode* root, int* returnSize) {
int* a = malloc(sizeof(int) * 501);
*returnSize = 0;
inorder(root, a, returnSize);//调用inorder函数
return a;
}
时间复杂度O(n);空间复杂度O(n)
【方法二】非递归算法
二叉树的中遍历按照:左-根-右来进行。
非递归通过栈来进行模拟。
第一步:创建一个栈。
第二步:定义一个stack来开辟一个栈空间。并将栈初始化,top=0。定义一个数组,用来返回最后中序遍历后的树,returnSize指针所指向的值初始化为0,定义一个移动指针p指向树的根,用来遍历树。
第三步:树不为空的时候 ,向左遍历。并入栈。左子树为空,出栈,并将出栈的元素加入数组a,然后右进行遍历右字树。最后返回数组a。
具体案例分析:
root = [1,null,2,3],输出:[1,3,2]。
开始时p指向root,也就是1,不为空,执行if(p != NULL)中的语句,将1进行入栈,将top++,也就是等于1,然后遍历左子树,此时p为空(1的左边),进行执行else里面的语句,先进行出栈,然后将top--,也就是等于0,然后将出栈的元素1加入数组a[0]中,并将(*returnSize)++。然后将p指向p的右子树,也就是2,此时p为2,不为空执行if(p != NULL)中的语句,将2进行入栈,将top++,也就是等于1,然后遍历左子树,此时左子树中p为3,不为空,执行if(p != NULL)中的语句,将3进行入栈(此时栈中为2->3),将top++,也就是等于2,然后遍历左子树,此时p为空(3的右边),进行执行else里面的语句,先进行出栈,然后将top--,也就是等于1,然后将出栈的元素3加入数组a[1]中,并将(*returnSize)++。然后将p指向p的右子树(3的右边),此时为空,进行执行else里面的语句,先进行出栈,然后将top--,也就是等于0,然后将出栈的元素2加入数组a[2]中,并将(*returnSize)++。然后将p指向p的右子树(2的右边),此时为空,进行执行else里面的语句,先进行出栈(栈为空),直接跳出while循环,返回数组a=[1,3,2]。
具体C语言代码如下。
//创建栈
struct stack{
struct TreeNode *s[100];
int top;
};
int* inorderTraversal(struct TreeNode* root, int* returnSize) {
//开辟一个栈空间
struct stack *stack=(struct stack*)malloc(sizeof(struct stack));
//初始化
stack->top=0;
//定义一个数组,用来返回最后中序遍历后的树
int *a=(int*)malloc(sizeof(int)*100);
//returnSize指针所指向的值初始化为0
*returnSize=0;
struct TreeNode *p=root;//定义一个移动指针指向树的根,用来遍历树
while(p != NULL || (stack->top) !=0){//树不为空或者栈不为空的时候,进行入栈和出栈操作
if(p != NULL){//树不为空的时候 ,向左遍历。并入栈
stack->s[stack->top]=p;
(stack->top)++;
p=p->left;
}
else{
//左子树为空,出栈,并将出栈的元素加入数组a,然后右进行遍历右字树
p=stack->s[(stack->top)-1];
(stack->top)--;
a[*returnSize]=p->val;
(*returnSize)++;
p=p->right;
}
}
return a;
}
时间复杂度O(n);空间复杂度O(h)