来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/count-complete-tree-nodes 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

    给你一棵 完全二叉树 的根节点 root ,求出该树的节点个数。 完全二叉树 的定义如下:在完全二叉树中,除了最底层节点可能没填满外,其余每层节点数都达到最大值,并且最下面一层的节点都集中在该层最左边的若干位置。若最底层为第 h 层,则该层包含 1~ 2h 个节点。

    1. /**
    2. * Definition for a binary tree node.
    3. * function TreeNode(val, left, right) {
    4. * this.val = (val===undefined ? 0 : val)
    5. * this.left = (left===undefined ? null : left)
    6. * this.right = (right===undefined ? null : right)
    7. * }
    8. */
    9. /**
    10. * @param {TreeNode} root
    11. * @return {number}
    12. */
    13. var countNodes = function(root) {
    14. if (!root) return 0;
    15. let stack = [root], count = 1;
    16. while (stack.length) {
    17. let temp = [];
    18. for (let item of stack) {
    19. if (item) {
    20. item.left && temp.push(item.left);
    21. item.right && temp.push(item.right);
    22. }
    23. }
    24. count += temp.length;
    25. stack = temp;
    26. }
    27. return count;
    28. };