Java實(shí)現(xiàn)求二叉樹的深度和寬度

2019-11-26 15:07:40

字體：大中小

來源：轉(zhuǎn)載

供稿：網(wǎng)友

這個(gè)是常見的對(duì)二叉樹的操作。總結(jié)一下：

設(shè)節(jié)點(diǎn)的數(shù)據(jù)結(jié)構(gòu)，如下：

復(fù)制代碼代碼如下:

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

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

1.二叉樹深度

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

復(fù)制代碼代碼如下:

// 獲取最大深度
    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.二叉樹寬度

　　使用隊(duì)列，層次遍歷二叉樹。在上一層遍歷完成后，下一層的所有節(jié)點(diǎn)已經(jīng)放到隊(duì)列中，此時(shí)隊(duì)列中的元素個(gè)數(shù)就是下一層的寬度。以此類推，依次遍歷下一層即可求出二叉樹的最大寬度。

復(fù)制代碼代碼如下:

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

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

        while (true) {
            int len = queue.size(); // 當(dāng)前層的節(jié)點(diǎn)個(gè)數(shù)
            if (len == 0)
                break;
            while (len > 0) {// 如果當(dāng)前層，還有節(jié)點(diǎn)
                TreeNode t = queue.poll();
                len--;
                if (t.left != null)
                    queue.add(t.left); // 下一層節(jié)點(diǎn)入隊(duì)
                if (t.right != null)
                    queue.add(t.right);// 下一層節(jié)點(diǎn)入隊(duì)
            }
            maxWitdth = Math.max(maxWitdth, queue.size());
        }
        return maxWitdth;
    }