简单编程题目连载（三）

二叉树中如何获取二叉树的层数？

注意这里获取的层数是从1开始算的，不是从0开始的。

直接上代码：

public class Node{    public int value;    public Node left;    public Node right;    public Node(int data){        this.value = data;    }}public void getHeight(Node h,int i){    if(h == null){        return i;    }    return Math.max(getHeight(h.left,i+1),getHeight(h.right,i+1));}

举个例子：

public class Test{    public static void main(String[] args){        Node node1 = new Node(1);        Node node2 = new Node(2);        Node node3 = new Node(3);        Node node4 = new Node(4);        Node node5 = new Node(5);        Node node6 = new Node(6);        Node node7 = new Node(7);        Node node8 = new Node(8);        Node node9 = new Node(9);        Node node10 = new Node(10);        Node node11 = new Node(11);        Node node12 = new Node(12);        Node node13 = new Node(13);        Node node14 = new Node(14);        Node node15 = new Node(15);        Node node16 = new Node(16);        node1.left = node2;        node1.right = node3;        node2.right = node4;        node4.left = node7;        node4.right = node8;        node8.right = node11;        node11.left = node13;        node11.right = node14;        node3.left = node5;        node3.right = node6;        node5.left = node9;        node5.right = node10;        node9.left = node12;        node12.left = node15;        node12.right = node16;        int height = getHeight(node1,0);    }}

上边的二叉树返回的高度为6。

如果获取二叉树每一层上最左和最右节点？

依然以上边的二叉树为例子

创建一个二维的Node类型的数组来存每层最左和最右的节点

Node[][] edgeMap = new Node[height][2];

数组的大小由二叉树的层数来定

然后利用下方的函数即可得到每层的最左和最右节点。

public void setEdgeMap(Node h,int i,Node[][] edgeMap){    if(h == null){        return;    }    edgeMap[i][0] = edgeMap[i][0] == null ? h : edgeMap[i][0];    edgeMap[i][1] = h;    setEdgeMap(h.left,i+1,edgeMap);    setEdgeMap(h.right,i+1,edgeMap);}

这个函数做了一件什么事呢？

二维数组的0位置作为左边，1位置作为右边。

i相当于到达的层数，每到一层，遇到的第一个左节点即为当前层的最左节点，因为该遍历是一直向左延伸的。右节点只要遇到新值就更新。然后递归遍历到每一层。递归基为遇到的节点为空，就返回。