题解 | #统计农场牛数量# java
统计农场牛数量
https://www.nowcoder.com/practice/c18924a6debf437180d77baec91dc586
import java.util.*; /* * public class TreeNode { * int val = 0; * TreeNode left = null; * TreeNode right = null; * public TreeNode(int val) { * this.val = val; * } * } */ public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param root TreeNode类 * @return int整型 */ int res = 0; public int countNodes (TreeNode root) { // write code here if (root == null) return 0; dfs(root, 1); return res; } private void dfs(TreeNode root, int cur) { if (root == null) return; res = Math.max(res, cur); dfs(root.left, cur << 1); dfs(root.right, (cur << 1) | 1); } }
该代码使用的编程语言是 Java。
这个题目主要涉及以下知识点:
- 二叉树的结构定义:定义了一个
TreeNode
结构体,包含一个整数值val
,以及左右子节点指针left
和right
。 - 二叉树节点的统计:统计二叉树的节点数量。
- 深度优先搜索(DFS):使用递归的方式遍历二叉树。
- 位运算:使用位移操作符
<<
和位或操作符|
进行二进制运算。
代码的文字解释大纲如下:
- 定义了一个
TreeNode
结构体,包含一个整数值val
,以及左右子节点指针left
和right
。 - 创建了一个
Solution
类,包含了统计二叉树节点数量的方法countNodes
。 countNodes
方法接收一个根节点root
,并返回二叉树的节点数量。如果根节点为空,直接返回0。使用一个整型变量 res 来统计节点数量,初始值为0。使用深度优先搜索(DFS)遍历二叉树,从根节点开始,每访问一个节点,将 res 加1。对于当前节点,递归访问其左子节点和右子节点,并通过位移操作符 << 和位或操作符 | 来计算子节点的位置。返回最终的节点数量 res。