力扣94:二叉树的中序遍历

给定一个二叉树的根节点 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)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值