获取二叉树的最大宽度(只能采用非递归的方式,采用堆的先进先出的方式去实现)

使用队列,层次遍历二叉树。在上一层遍历完成后,下一层的所有节点已经放到队列中,此时队列中的元素个数就是下一层的宽度。以此类推,依次遍历下一层即可求出二叉树的最大宽度。 

// 获取最大宽度
     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;
     }

猜你喜欢

转载自blog.csdn.net/shujuelin/article/details/86491757