144. 二叉树的前序遍历 - 力扣(LeetCode)https://leetcode.cn/problems/binary-tree-preorder-traversal/description/
/**
* 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_size(struct TreeNode* root)
{
return root == NULL?0:preorderTraversal_size(root->left)+preorderTraversal_size(root->right)+1;
}
//前序遍历,放到数组里面
void preorder(struct TreeNode* root,int* a,int* pi)
{
//结束递归
if(root==NULL)
{
return;
}
a[(*pi)++] = root->val;
preorder(root->left,a,pi);
preorder(root->right,a,pi);
}
//输出函数
int* preorderTraversal(struct TreeNode* root, int* returnSize)
{
*returnSize = preorderTraversal_size(root);
int* a = (int*)malloc(sizeof(int)*(*returnSize));
if(a == NULL)
{
perror("preorder:a == NULL");
exit(1);
}
int i = 0;
preorder(root,a,&i);
return a;
}
解释:
-
int preorderTraversal_size(struct TreeNode* root)
- 这个函数用于计算二叉树中节点的数量。
- 如果根节点
root
为空,返回0
。 - 否则,递归地计算左子树和右子树的节点数量,然后加
1
(包括根节点)。
-
void preorder(struct TreeNode* root, int* a, int* pi)
- 这个函数用于前序遍历二叉树,并将节点值存储到数组
a
中。 - 如果根节点
root
为空,递归结束。 - 否则,将根节点的值赋给数组
a
的第*pi
个位置,并将pi
递增。 - 递归地对左子树和右子树执行相同的操作。
- 这个函数用于前序遍历二叉树,并将节点值存储到数组
-
int* preorderTraversal(struct TreeNode* root, int* returnSize)
- 这个函数用于初始化前序遍历的过程,并返回一个包含遍历结果的新分配的数组。
*returnSize
被设置为preorderTraversal_size(root)
的返回值,即树中节点的数量。- 为数组分配内存,大小为
*returnSize
。 - 如果内存分配失败,输出错误并退出程序。
- 初始化一个计数器
i
,用于跟踪数组中当前的位置。 - 调用
preorder
函数,开始前序遍历过程,并将节点值存入数组。 - 返回包含前序遍历结果的数组。
总结
preorderTraversal_size
函数通过递归计算二叉树的节点数量。preorder
函数通过递归执行前序遍历,将每个访问到的节点值存入数组。preorderTraversal
函数初始化遍历过程,计算所需的内存大小,分配内存,并调用preorder
函数来填充数组。最终返回一个包含树的前序遍历结果的数组。