【二叉树】LeetCode.144:二叉树的前序遍历(小细节把握)

阅读 5

2024-05-26

🎁个人主页:我们的五年

🔍系列专栏:初阶初阶结构刷题

🎉欢迎大家点赞👍评论📝收藏⭐文章

 

目录

1.题目描述:​编辑

2.问题分析:

🍔函数解读:

🍔确定数组的大小:

🍔调用遍历函数:

3.最终代码:


 

1.题目描述:

2.问题分析:

🍔函数解读:

🍔确定数组的大小:

我们要去申请数组,那就要确定数组的大小,那么我们把数组的大小顶为多大呢?

题中说了节点数目在[0,100]。


📷给出下面几种情况进行选择:(看看哪种情况最好)

1.因为题目中给了最多为100个节点,所以申请100*sizeof(int)的大小?

2.先申请小一点,不够的话就再去扩容?

3.先去计算树的大小,再去扩容?

分析:

1.如果题目给的是一亿个,我们不可能去申请一亿大小的空间。而且这种情况会有空间浪费。

2.如果频繁的扩容会造成速度很慢,特别是异地扩容,realloc内部自己还要去移动数据。

3.情况三不会浪费空间,又不会频繁扩容。


所以我们先去计算树的节点个数:

分治思想:每次都把树分成三个部分,根+左子树+右子树

最小子问题根为NULL就返回0.

 int TreeSize(struct TreeNode* root)
 {
    if(root==NULL)
        return 0;
    //分治,总个数等于根+加左子树的个数+右子树的个数
    return TreeSize(root->left)+TreeSize(root->right)+1;
 }

🍔调用遍历函数:

测试用例过了一半,在哪个测试用例就过不了呢?

运行测试用例都能过

如果细心一点的就可以发现,左右子树都有的时候,就过不了,只有一边有树,或者只有根就可以过。

因为这样左右子树开始时都是一样的,如果这样,调用右边的时候,又把左边已经覆盖的值又去覆盖了一遍,所以左边子树的值就没了。

3.最终代码:

/**
 * 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)
 {
    if(root==NULL)
        return 0;
    return TreeSize(root->left)+TreeSize(root->right)+1;
 }

void preorder(struct TreeNode* root,int* a,int i)
{
    if(root==NULL)
        return;
    a[i++]=root->val;
    preorder(root->left,a,i);
    preorder(root->right,a,i);
}

int* preorderTraversal(struct TreeNode* root, int* returnSize) {
    *returnSize=TreeSize(root);
    int *a=(int*)malloc(sizeof(int)*(*returnSize));
    int i=0;
    preorder(root,a,i);
    return a;
}

精彩评论(0)

0 0 举报