【C语言题解】 | 144. 二叉树的前序遍历

144. 二叉树的前序遍历

提示：

1. 树中节点数目在范围 [0, 100] 内

函数原型：

int* preorderTraversal(struct TreeNode* root, int* returnSize) {
   

首先先观察一下这个函数原型，TreeNode* root 为形参，传入根节点，int* returnSize为形参，在函数调用时用于返回改题目所求数组的长度，因为由于C语言的局限，只能返回一个参数，所以采用这种通过传入指针的形参，来改变函数外部实参的方法。

题目要求给一个二叉树的根节点，返回其前序遍历的数组。

首先先计算二叉树的节点个数，用于后续的数组空间申请。

int TreeSize(struct TreeNode* root)
 {
   
     return root == NULL ? 0 : TreeSize(root->left) + TreeSize(root->right) + 1;
 }

然后先序遍历，写入数组：

因为根据上述代码，求得节点个数为n，则该数组一共有n个空间，控制写入数组的下标需要传入int* ，因为若直接传入int，形参的改变不影响实参的改变。

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

使用malloc函数构建数组，返回数组。

int* preorderTraversal(struct TreeNode* root, int* returnSize) {
   
    int n = TreeSize(root);
    int* a = (int*)malloc(sizeof(int)*n);
    *returnSize = n;

    int* i = 0;
    preorder(root,a,&i);
    return a;
}

代码

/**
 * 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 TreeSize(struct TreeNode* root)
 {
   
     return root == NULL ? 0 : TreeSize(root->left) + TreeSize(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) {
   
    int n = TreeSize(root);
    int* a = (int*)malloc(sizeof(int)*n);
    *returnSize = n;

    int* i = 0;
    preorder(root,a,&i);
    return a;
}