求二叉樹的深度和寬度[Java]

求二叉樹的深度和寬度[Java]

這個是常見的對二叉樹的操作。總結一下：

設節點的數據結構，如下：

class TreeNode {
    char val;
    TreeNode left = null;
    TreeNode right = null;

    TreeNode(char _val) {
        this.val = _val;
    }
}

1.二叉樹深度

　　這個可以使用遞歸，分別求出左子樹的深度、右子樹的深度，兩個深度的較大值+1即可。

      基本運算是求出最下面一個父節點的左右子樹是否存在，存在就爲1，返回1，不存在就返回0。再向上遞歸，每次累加

// 獲取最大深度
    public static int getMaxDepth(TreeNode root) {
        if (root == null)
            return 0;
        else {
            int left = getMaxDepth(root.left);
            int right = getMaxDepth(root.right);
            return 1 + Math.max(left, right);
        }
    }

2.二叉樹寬度

     相當於層次遍歷，要用輔助數據結構，隊列，因爲會涉及到左子樹右子樹之間的跳轉（數據結構書155頁）

　　使用隊列，層次遍歷二叉樹。在上一層遍歷完成後，下一層的所有節點已經放到隊列中，此時隊列中的元素個數就是下一層的寬度。以此類推，依次遍歷下一層即可求出二叉樹的最大寬度。

// 獲取最大寬度
    public static int getMaxWidth(TreeNode root) {
        if (root == null)
            return 0;

        Queue<TreeNode> queue = new ArrayDeque<TreeNode>();
        int maxWitdth = 1; // 最大寬度
        queue.add(root); // 入隊

        while (true) {
            int len = queue.size(); // 當前層的節點個數
            if (len == 0)
                break;
            while (len > 0) {// 如果當前層，還有節點
                TreeNode t = queue.poll();
                len--;
                if (t.left != null)
                    queue.add(t.left); // 下一層節點入隊
                if (t.right != null)
                    queue.add(t.right);// 下一層節點入隊
            }
            maxWitdth = Math.max(maxWitdth, queue.size());
        }
        return maxWitdth;
    }