二叉树习题精讲- 二叉树的前序遍历

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;
}

解释：

1. int preorderTraversal_size(struct TreeNode* root)

   • 这个函数用于计算二叉树中节点的数量。
   • 如果根节点 root 为空，返回 0。
   • 否则，递归地计算左子树和右子树的节点数量，然后加 1（包括根节点）。

2. void preorder(struct TreeNode* root, int* a, int* pi)

   • 这个函数用于前序遍历二叉树，并将节点值存储到数组 a 中。
   • 如果根节点 root 为空，递归结束。
   • 否则，将根节点的值赋给数组 a 的第 *pi 个位置，并将 pi 递增。
   • 递归地对左子树和右子树执行相同的操作。

3. int* preorderTraversal(struct TreeNode* root, int* returnSize)

   • 这个函数用于初始化前序遍历的过程，并返回一个包含遍历结果的新分配的数组。
   • *returnSize 被设置为 preorderTraversal_size(root) 的返回值，即树中节点的数量。
   • 为数组分配内存，大小为 *returnSize。
   • 如果内存分配失败，输出错误并退出程序。
   • 初始化一个计数器 i，用于跟踪数组中当前的位置。
   • 调用 preorder 函数，开始前序遍历过程，并将节点值存入数组。
   • 返回包含前序遍历结果的数组。

总结

• preorderTraversal_size 函数通过递归计算二叉树的节点数量。
• preorder 函数通过递归执行前序遍历，将每个访问到的节点值存入数组。
• preorderTraversal 函数初始化遍历过程，计算所需的内存大小，分配内存，并调用 preorder 函数来填充数组。最终返回一个包含树的前序遍历结果的数组。