求完全二叉树节点的个数,要求时间复杂度低于O(N)

最新推荐文章于 2024-07-19 17:08:05 发布

朂後哋箹萣

最新推荐文章于 2024-07-19 17:08:05 发布

阅读量144

点赞数

分类专栏： LeetCode 文章标签：完全二叉树时间复杂度算法节点计数二叉树遍历

本文链接：https://blog.csdn.net/z1171127310/article/details/127837114

版权

LeetCode 专栏收录该内容

85 篇文章 4 订阅

订阅专栏

问题描述

求完全二叉树节点的个数,要求时间复杂度低于O(N)。

思想

首先先求完全二叉树的高度h，然后再求右子树的高度。
如果右子树的高度level小于完全二叉树的高h，那么右树是一颗满二叉树，右树节点个数为：2^(h-level-1)-1。
如果右子树的高度level等于完全二叉树的高h，那么左树是一颗满二叉树，左树节点个数为：2^(h-level)-1。

算法的时间复杂度为：O((logN)^2) 。

代码

    public static class Node {
        public int value;
        public Node left;
        public Node right;

        public Node(int data) {
            this.value = data;
        }
    }

    public static int nodeNum(Node head) {
        if (head == null) {
            return 0;
        }
        return bs(head, 1, mostLeftLevel(head, 1));
    }

    public static int bs(Node node, int Level, int h) {
        if (Level == h) {
            return 1;
        }
        if (mostLeftLevel(node.right, Level + 1) == h) {
            return (1 << (h - Level)) + bs(node.right, Level + 1, h);
        } else {
            return (1 << (h - Level - 1)) + bs(node.left, Level + 1, h);
        }
    }

    public static int mostLeftLevel(Node node, int level) {
        while (node != null) {
            level++;
            node = node.left;
        }
        return level - 1;
    }

    public static void main(String[] args) {
        Node head = new Node(1);
        head.left = new Node(2);
        head.right = new Node(3);
        head.left.left = new Node(4);
        head.left.right = new Node(5);
        head.right.left = new Node(6);
        System.out.println(nodeNum(head));
    }