leetcode 二叉树习题---开幕式烟火

老鱼37

于 2022-09-13 20:05:36 发布

阅读量107

点赞数 1

分类专栏：暑假习题文章标签： leetcode

本文链接：https://blog.csdn.net/qq_61342044/article/details/126840585

版权

暑假习题专栏收录该内容

17 篇文章 0 订阅

订阅专栏

在这里插入图片描述
思路：
用一个数组计数，全部memset为0 ，然后前序遍历二叉树计数，最后再遍历一遍数组，然后不为0 就+1

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    void preorder(int *hash,TreeNode*root2)
    {
        if(root2)//不为空
        {
            hash[root2->val]++;
            preorder(hash,root2->right);
            preorder(hash,root2->left);
        }
    }
public:
    int numColor(TreeNode* root) {
        //用数组计数的方法
        int hash[1001];
        memset(hash,0,sizeof(hash));
        int count=0;
        preorder(hash,root);//前序遍历
        for(int i=1;i<1001;i++)
        {
            if(hash[i])
                count++;
        }
        return count;
    }
};