【算法学习】LCP 44

2401_84692110

于 2024-04-29 15:51:51 发布

阅读量100

点赞数 3

分类专栏：程序员文章标签：算法学习深度优先

本文链接：https://blog.csdn.net/2401_84692110/article/details/138315758

版权

程序员专栏收录该内容

110 篇文章 0 订阅

订阅专栏

输入：
	root = [1,3,2,1,null,2]
输出：
	3
解释：
	焰火中有 3 个不同的颜色，值分别为 1、2、3

样例 2

输入：
	root = [3,3,3]
输出：
	1
解释：
	焰火中仅出现 1 个颜色，值为 3

提示

1 <= 节点个数 <= 1000
1 <= Node.val <= 1000

分析

翻译一下题意就是看整个树里一共有几种不同的值。
所以考察了2个方面的知识点，一个是二叉树数据结构，一个是统计计数。
最容易想到的是用Set之类的数据结构。
循环和递归都可以。
提示中已经给定了节点值的范围，所以可以使用数组这种底层数据结构去替代Set等数据结构。
在遍历树的过程中判断计数，或者在最后再对计数的数据结构遍历计数，这两种方式都可以，在遍历树中判断计数受节点的数量影响，在最后再遍历计数数据结构计数受节点值取值范围影响。

题解

java

/\*\*
 \* Definition for a binary tree node.
 \* public class TreeNode {
 \* int val;
 \* TreeNode left;
 \* TreeNode right;
 \* TreeNode(int x) { val = x; }
 \* }
 \*/
class Solution {
    public int numColor(TreeNode root) {
		int ans = 0;
		
		boolean[] flag = new boolean[1001];
		dfs(root, flag);
		for (boolean f : flag) {
			if (f) {
				ans++;
			}
		}
		
		return ans;
	}

	private void dfs(TreeNode root, boolean[] flag) {
		if (root != null) {
			flag[root.val] = true;
			dfs(root.left, flag);
			dfs(root.right, flag);
		}
	}
}

c

/\*\*
 \* Definition for a binary tree node.
 \* struct TreeNode {
 \* int val;
 \* struct TreeNode \*left;
 \* struct TreeNode \*right;
 \* };
 \*/

int numColor(struct TreeNode \*root) {
    int ans = 0;

    bool flag[1001] = {false};
    dfs(root, flag);
    for (int i = 1; i < 1001; ++i) {
        if (flag[i]) {
            ans++;
        }
    }

    return ans;
}

void dfs(struct TreeNode \*root, bool \*flag) {
    if (root) {
        flag[root->val] = true;
        dfs(root->left, flag);
        dfs(root->right, flag);
    }
}

c++

/\*\*
 \* 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 {
public:
    int numColor(struct TreeNode \*root) {
        int ans = 0;
        
        bool flag[1001] = {false};
        dfs(root, flag);
        for (bool f : flag) {
            if (f) {
                ans++;
            }
        }
        
        return ans;
    }

    void dfs(struct TreeNode \*root, bool \*flag) {
        if (root != nullptr) {
            flag[root->val] = true;
            dfs(root->left, flag);
            dfs(root->right, flag);
        }
    }
};

python

# Definition for a binary tree node.
# class TreeNode:
# def \_\_init\_\_(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
    def numColor(self, root: TreeNode) -> int:
        ans = 0

        flag = [False] \* 1001

        def dfs(n):
            if n:
                flag[n.val] = True
                dfs(n.left)
                dfs(n.right)

        dfs(root)

        for f in flag:
            if f:
                ans += 1

        return ans

go

/\*\*
 \* Definition for a binary tree node.
 \* type TreeNode struct {
 \* Val int
 \* Left \*TreeNode
 \* Right \*TreeNode
 \* }
 \*/
func numColor(root \*TreeNode) int {
    ans := 0



![在这里插入图片描述](https://img-blog.csdnimg.cn/20210511152217670.jpg?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3poaWd1aWd1,size_16,color_FFFFFF,t_70)

**感谢每一个认真阅读我文章的人，看着粉丝一路的上涨和关注，礼尚往来总是要有的：**



①　2000多本Python电子书（主流和经典的书籍应该都有了）

②　Python标准库资料（最全中文版）

③　项目源码（四五十个有趣且经典的练手项目及源码）

④　Python基础入门、爬虫、web开发、大数据分析方面的视频（适合小白学习）

⑤ Python学习路线图（告别不入流的学习）




**网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

**[需要这份系统化学习资料的朋友，可以戳这里无偿获取](https://bbs.csdn.net/topics/618317507)**

**一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！**