解题思路:递归
1.确定递归函数的参数和返回值:参数就是传入树的根节点,返回就返回以该节点为根节点二叉树的节点数量,所以返回值为int类型。
int countNodes(TreeNode root)
2.确定终止条件:如果为空节点的话,就返回0,表示节点数为0。
if(root == null) {
return 0;
}
3.确定单层递归的逻辑:先求它的左子树的节点数量,再求的右子树的节点数量,最后取总和再加一 (加1是因为算上当前中间节点)就是目前节点为根节点的节点数量。
int leftNum = countNodes(root.left); // 左
int rightNum =countNodes(root.right); // 右
int treeNum = leftNum + rightNum + 1; // 中
return treeNum;
综上,本题代码如下:
class Solution {
// 通用递归解法
public int countNodes(TreeNode root) {
if(root == null) {
return 0;
}
int leftNum = countNodes(root.left); // 左
int rightNum =countNodes(root.right); // 右
int treeNum = leftNum + rightNum + 1; // 中
return treeNum;
}
}