c#二叉树 取叶子节点个数_LeetCode 222. 完全二叉树节点个数

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

完全二叉树（complete Binary Tree）： 完全二叉树是由满二叉树而引出来的。 对于深度为K的，有n个结点的二叉树，当且仅当其每一个结点都与深度为K的满二叉树中编号从1至n的结点一一对应时称之为完全二叉树。

注意：在中文的满二叉树对应英文中的Perfect Binary Tree，英文中的Full Binary Tree指一棵二叉树的所有节点要么没有孩子节点，要么有两个孩子节点。

由此可知，完全二叉树的叶子节点要么在最后一层，要么在倒数第二层。

如果求一颗普通二叉树的节点个数，采用递归做法，时间复杂度为O（N）。

/**
* 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 countNodes(TreeNode* root) {
        if(root == NULL)
            return 0;
        return 1 + countNodes(root->left) + countNodes(root->right);
    }
};

如果求一颗满二叉树的节点个数，可以先计算树的高度，按照公式求得节点数，时间复杂度为O（logN）。

/**
* 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 countNodes(TreeNode* root) {
        if(root == NULL)
            return 0;
        int h = 0;
        while(root != NULL) {
            h ++;
            root = root->left;
        }
        return (2 << h - 1) - 1;
    }
};

求完全二叉树节点个数结合了以上两种方法，首先计算当前树左右子树的树高，判断当前树是否为完全二叉树。如果是完全二叉树，则公式计算，否则递归计算左右子树。

复杂度分析:对于一个完全二叉树的节点做左右的递归，由下图分析可知，这两个递归只有一个会递归下去，因为 一棵完全二叉树的两棵子树，至少有一棵是满二叉树。

那么，算法的递归深度就是树的高度O（logN）,每次递归花费的时间为O（logN）（while循环的时间），那么总体的时间复杂度为O（logN*logN）。

/**
* 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 countNodes(TreeNode* root) {
        if(root == NULL)
            return 0;
        TreeNode* left_r = root, *right_r = root;
        int l_h = 0, r_h = 0;
        while(left_r != NULL) {
            l_h ++;
            left_r = left_r->left;
        }
        while(right_r != NULL) {
            r_h ++;
            right_r = right_r->right;
        }
        if(l_h == r_h) {
            return (2 << l_h - 1) - 1;
        }
        return 1 + countNodes(root->left) + countNodes(root->right);
    }
};