Java 题解 | #统计农场牛数量#

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

    public 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。、

这道题考察的知识点是二叉树的遍历和节点计数。

代码解释如下：

1. 定义了一个结构体 TreeNode，表示二叉树节点。每个节点包含一个整数值 val，以及左子节点 left 和右子节点 right。
2. 定义了一个类 Solution，其中包含了 countNodes 方法。
3. countNodes 方法接收一个二叉树的根节点 root 作为参数。首先判断根节点是否为空，如果为空则返回0。
4. 定义了一个整型变量 res 来记录节点的数量。
5. 调用辅助方法 dfs 进行深度优先搜索（DFS），同时传入当前节点和当前节点对应的编号 cur。初始时，cur 被设置为1。
6. 在 dfs 方法中，如果当前节点为空，则直接返回。否则，将 cur 和 res 的较大值赋给 res。
7. 继续递归调用 dfs 方法，分别传入当前节点的左子节点和右子节点，并更新 cur 的值。对于左子节点，将 cur 左移一位（cur << 1）。对于右子节点，将 cur 左移一位后加上1（(cur << 1) | 1）。
8. 最终，res 中记录的就是二叉树中的节点数量。

该代码实现了统计二叉树中节点的数量的功能。通过深度优先搜索遍历二叉树的每个节点，并使用一个全局变量 res 记录节点的数量。每次访问一个节点时，将其对应的编号 cur 与 res 的较大值赋给 res。最终，res 中存储的就是二叉树中节点的数量。