溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

如何進(jìn)行python二叉樹的層次遍歷

發(fā)布時(shí)間:2021-12-13 15:15:33 來源:億速云 閱讀:260 作者:柒染 欄目:大數(shù)據(jù)

本篇文章給大家分享的是有關(guān)如何進(jìn)行python二叉樹的層次遍歷,小編覺得挺實(shí)用的,因此分享給大家學(xué)習(xí),希望大家閱讀完這篇文章后可以有所收獲,話不多說,跟著小編一起來看看吧。

給定一個(gè)二叉樹,返回其按層次遍歷的節(jié)點(diǎn)值。(即逐層地,從左到右訪問所有節(jié)點(diǎn))。

例如:

給定二叉樹: [3,9,20,null,null,15,7],

    3

   / \

  9  20

    /  \

   15   7

返回其層次遍歷結(jié)果:

[

  [3],

  [9,20],

  [15,7]

]

答案:

 1public List<List<Integer>> levelOrder1(TreeNode root) {
2    Queue<TreeNode> queue = new LinkedList<TreeNode>();
3    List<List<Integer>> wrapList = new LinkedList<List<Integer>>();
4    if (root == null)
5        return wrapList;
6    queue.offer(root);
7    while (!queue.isEmpty()) {
8        int levelNum = queue.size();
9        List<Integer> subList = new LinkedList<Integer>();
10        for (int i = 0; i < levelNum; i++) {
11            if (queue.peek().left != null)
12                queue.offer(queue.peek().left);
13            if (queue.peek().right != null)
14                queue.offer(queue.peek().right);
15            subList.add(queue.poll().val);
16        }
17        wrapList.add(subList);
18    }
19    return wrapList;
20}

解析:

LinkedList是個(gè)鏈表,先進(jìn)先出,levelNum是每一層的節(jié)點(diǎn)數(shù)量,一層一層的遍歷,代碼沒什么難度,下面再來看一種遞歸的寫法

 1public List<List<Integer>> levelOrder(TreeNode root) {
2    List<List<Integer>> res = new ArrayList<List<Integer>>();
3    levelHelper(res, root, 0);
4    return res;
5}
6
7public void levelHelper(List<List<Integer>> res, TreeNode root, int height) {
8    if (root == null) return;
9    if (height >= res.size()) {
10        res.add(new LinkedList<Integer>());
11    }
12    res.get(height).add(root.val);
13    levelHelper(res, root.left, height + 1);
14    levelHelper(res, root.right, height + 1);
15}

height其實(shí)相當(dāng)于層級(jí),height>=res.size()說明已經(jīng)到了下一層了,所以要新建一個(gè)list。

以上就是如何進(jìn)行python二叉樹的層次遍歷,小編相信有部分知識(shí)點(diǎn)可能是我們?nèi)粘9ぷ鲿?huì)見到或用到的。希望你能通過這篇文章學(xué)到更多知識(shí)。更多詳情敬請(qǐng)關(guān)注億速云行業(yè)資訊頻道。

向AI問一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場,如果涉及侵權(quán)請(qǐng)聯(lián)系站長郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI