简单编程题目连载（三）

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

注意这里获取的层数是从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相当于到达的层数，每到一层，遇到的第一个左节点即为当前层的最左节点，因为该遍历是一直向左延伸的。右节点只要遇到新值就更新。然后递归遍历到每一层。递归基为遇到的节点为空，就返回。