leetcode 222 学习笔记（求节点个数，递归迭代）

222. 完全二叉树的节点个数

问题描述

传送门

思路

依然可以使用递归法和迭代法来解决。

这道题目的递归法和求二叉树的深度写法类似。

而迭代法中，二叉树的层序遍历模板稍稍修改一下，记录遍历的节点数量就可以了。
递归遍历的顺序依然是后序（左右中）。

1、递归法

①确定递归函数的参数和返回值：
参数就是传入树的根节点，返回就返回以该节点为根节点二叉树的节点数量，所以返回值为int类型。

	int getNodesNum(TreeNode* cur) {

②确定终止条件：
如果为空节点的话，就返回0，表示节点数为0。

	if (cur == NULL) return 0;

③确定单层递归的逻辑：
先求它的左子树的节点数量，再求它的右子树的节点数量，最后取总和再加一 （加1是因为算上当前中间节点）就是目前节点为根节点的节点数量。

	int leftNum = getNodesNum(cur->left);      // 左
	int rightNum = getNodesNum(cur->right);    // 右
	int treeNum = leftNum + rightNum + 1;      // 中
	return treeNum;

整体代码如下:

	class Solution {
	private:
	    int getNodesNum(TreeNode* cur) {
	        if (cur == 0) return 0;
	        int leftNum = getNodesNum(cur->left);      // 左
	        int rightNum = getNodesNum(cur->right);    // 右
	        int treeNum = leftNum + rightNum + 1;      // 中
	        return treeNum;
	    }
	public:
	    int countNodes(TreeNode* root) {
	        return getNodesNum(root);
	    }
};

精简代码如下:

	class Solution {
	public:
	    int countNodes(TreeNode* root) {
	        if (root == NULL) return 0;
	        return 1 + countNodes(root->left) + countNodes(root->right);
	    }
	};

2、迭代法

只要模板稍做改动，加一个变量result，统计节点数量就可以了

	class Solution {
	public:
	    int countNodes(TreeNode* root) {
	        queue<TreeNode*> que;
	        if (root != NULL) 
	        	que.push(root);
	        int result = 0;
	        while (!que.empty()) {
	            int size = que.size();
	            for (int i = 0; i < size; i++) {
	                TreeNode* node = que.front();
	                que.pop();
	                result++;   // 记录节点数量
	                if (node->left) que.push(node->left);
	                if (node->right) que.push(node->right);
	            }
	        }
	        return result;
	    }
	};