数据结构与算法之完全二叉树的节点个数
目录
- 数据结构与算法之完全二叉树的节点个数
1. 数据结构与算法之完全二叉树的节点个数
(一)思路
- 先遍历左边界获得树的高度h
- 遍历右子树的左边界,看看到没到最后一层,如果到了,那么左子树是满二叉树,右子树也可以这样分析
- 同理,遍历右子树的左边界,如果没有到最后一层,那么右子树是满二叉树,左子树也可以这样分析。
(二) 代码实现
public class Code_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) {
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));
}
}