前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >已知一棵完全二叉树,求其节点的个数

已知一棵完全二叉树,求其节点的个数

作者头像
名字是乱打的
发布2022-05-13 09:52:24
1410
发布2022-05-13 09:52:24
举报
文章被收录于专栏:软件工程

已知一棵完全二叉树,求其节点的个数 要求:时间复杂度低于O(N),N为这棵树的节点个数

代码

代码语言:javascript
复制
package com.algorithm.practice.tree;

public class CompleteTreeNodeNumber {

    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) { //以当前node结点为头的树有多少个结点
        if (level == h) { //node为当前结点,level为该结点在第几层,h为固定值,其实就是整个树的高度
            return 1;
        }
        if (mostLeftLevel(node.right, level + 1) == h) {//如果当前右子树的最左结点到了最后一层
            return (1 << (h - level)) + bs(node.right, level + 1, h);//当前结点左子树结点树+bs右结点
        } else {//如果当前结点右子树最左结点没有达到最后一层,那么右子树是一个满二叉树,且高度比左子树少1
            return (1 << (h - level - 1)) + bs(node.left, level + 1, h);//当前结点右子树结点树+bs左结点
        }
    }

    public static int mostLeftLevel(Node node, int level) {//返回node的最左结点到了哪一层
        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));

    }

}

时间复杂度

每次遍历查找一个结点,次数是深度logN 最左结点个数是logN 那么总时间复杂度是O(logN^2)

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022-05-13,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 代码
  • 时间复杂度
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档