从左到右打印二叉树的所有叶节点 | Set-2(迭代方法)
给定一个二叉树,任务是从左到右打印叶子节点。节点必须按照它们从左到右出现的顺序打印。
例子:
Input :
1
/ \
2 3
/ \ / \
4 5 6 7
Output :4 5 6 7
Input :
4
/ \
5 9
/ \ / \
8 3 7 2
/ / \
12 6 1
Output :12 3 7 6 1
我们已经讨论了递归方法。在这里,我们将使用两个堆栈来解决问题。
方法:想法是使用两个堆栈,一个存储树的所有节点,另一个存储所有叶子节点。我们将弹出第一个堆栈的顶部节点。如果节点有左孩子,我们将它压入第一个堆栈的顶部,如果它有一个右孩子,我们将它压入第一个堆栈的顶部,但如果节点是叶节点,我们将推入它放在第二个堆栈的顶部。我们将对所有节点执行此操作,直到我们完全遍历二叉树。
然后我们将开始弹出第二个堆栈并打印其所有元素,直到堆栈变空。
下面是上述方法的实现:
C++
// C++ program to print all the leaf nodes
// of a Binary tree from left to right
#include
using namespace std;
// Binary tree node
struct Node {
Node* left;
Node* right;
int data;
};
// Function to create a new
// Binary node
Node* newNode(int data)
{
Node* temp = new Node;
temp->data = data;
temp->left = NULL;
temp->right = NULL;
return temp;
}
// Function to Print all the leaf nodes
// of Binary tree using two stacks
void PrintLeafLeftToRight(Node* root)
{
// Stack to store all the nodes of tree
stack s1;
// Stack to store all the leaf nodes
stack s2;
// Push the root node
s1.push(root);
while (!s1.empty()) {
Node* curr = s1.top();
s1.pop();
// If current node has a left child
// push it onto the first stack
if (curr->left)
s1.push(curr->left);
// If current node has a right child
// push it onto the first stack
if (curr->right)
s1.push(curr->right);
// If current node is a leaf node
// push it onto the second stack
else if (!curr->left && !curr->right)
s2.push(curr);
}
// Print all the leaf nodes
while (!s2.empty()) {
cout << s2.top()->data << " ";
s2.pop();
}
}
// Driver code
int main()
{
Node* root = newNode(1);
root->left = newNode(2);
root->right = newNode(3);
root->left->left = newNode(4);
root->right->left = newNode(5);
root->right->right = newNode(7);
root->left->left->left = newNode(10);
root->left->left->right = newNode(11);
root->right->right->left = newNode(8);
PrintLeafLeftToRight(root);
return 0;
}
Java
// Java program to print all the leaf nodes
// of a Binary tree from left to right
import java.util.*;
class GFG
{
// Binary tree node
static class Node
{
Node left;
Node right;
int data;
};
// Function to create a new
// Binary node
static Node newNode(int data)
{
Node temp = new Node();
temp.data = data;
temp.left = null;
temp.right = null;
return temp;
}
// Function to Print all the leaf nodes
// of Binary tree using two stacks
static void PrintLeafLeftToRight(Node root)
{
// Stack to store all the nodes of tree
Stack s1 = new Stack<>();
// Stack to store all the leaf nodes
Stack s2 = new Stack<>();;
// Push the root node
s1.push(root);
while (!s1.empty())
{
Node curr = s1.peek();
s1.pop();
// If current node has a left child
// push it onto the first stack
if (curr.left!=null)
s1.push(curr.left);
// If current node has a right child
// push it onto the first stack
if (curr.right!=null)
s1.push(curr.right);
// If current node is a leaf node
// push it onto the second stack
else if (curr.left==null && curr.right==null)
s2.push(curr);
}
// Print all the leaf nodes
while (!s2.empty())
{
System.out.print(s2.peek().data +" ");
s2.pop();
}
}
// Driver code
public static void main(String[] args)
{
Node root = newNode(1);
root.left = newNode(2);
root.right = newNode(3);
root.left.left = newNode(4);
root.right.left = newNode(5);
root.right.right = newNode(7);
root.left.left.left = newNode(10);
root.left.left.right = newNode(11);
root.right.right.left = newNode(8);
PrintLeafLeftToRight(root);
}
}
// This code is contributed by 29AjayKumar
Python3
# Python3 program to print all the leaf
# nodes of a Binary tree from left to right
# Binary tree node
class Node:
def __init__(self, data):
self.data = data
self.left = None
self.right = None
# Function to Print all the leaf nodes
# of Binary tree using two stacks
def PrintLeafLeftToRight(root):
# Stack to store all the nodes
# of tree
s1 = []
# Stack to store all the
# leaf nodes
s2 = []
# Push the root node
s1.append(root)
while len(s1) != 0:
curr = s1.pop()
# If current node has a left child
# push it onto the first stack
if curr.left:
s1.append(curr.left)
# If current node has a right child
# push it onto the first stack
if curr.right:
s1.append(curr.right)
# If current node is a leaf node
# push it onto the second stack
elif not curr.left and not curr.right:
s2.append(curr)
# Print all the leaf nodes
while len(s2) != 0:
print(s2.pop().data, end = " ")
# Driver code
if __name__ == "__main__":
root = Node(1)
root.left = Node(2)
root.right = Node(3)
root.left.left = Node(4)
root.right.left = Node(5)
root.right.right = Node(7)
root.left.left.left = Node(10)
root.left.left.right = Node(11)
root.right.right.left = Node(8)
PrintLeafLeftToRight(root)
# This code is contributed
# by Rituraj Jain
C#
// C# program to print all the leaf nodes
// of a Binary tree from left to right
using System;
using System.Collections.Generic;
class GFG
{
// Binary tree node
public class Node
{
public Node left;
public Node right;
public int data;
};
// Function to create a new
// Binary node
static Node newNode(int data)
{
Node temp = new Node();
temp.data = data;
temp.left = null;
temp.right = null;
return temp;
}
// Function to Print all the leaf nodes
// of Binary tree using two stacks
static void PrintLeafLeftToRight(Node root)
{
// Stack to store all the nodes of tree
Stack s1 = new Stack();
// Stack to store all the leaf nodes
Stack s2 = new Stack();;
// Push the root node
s1.Push(root);
while (s1.Count != 0)
{
Node curr = s1.Peek();
s1.Pop();
// If current node has a left child
// push it onto the first stack
if (curr.left != null)
s1.Push(curr.left);
// If current node has a right child
// push it onto the first stack
if (curr.right != null)
s1.Push(curr.right);
// If current node is a leaf node
// push it onto the second stack
else if (curr.left == null && curr.right == null)
s2.Push(curr);
}
// Print all the leaf nodes
while (s2.Count != 0)
{
Console.Write(s2.Peek().data + " ");
s2.Pop();
}
}
// Driver code
public static void Main(String[] args)
{
Node root = newNode(1);
root.left = newNode(2);
root.right = newNode(3);
root.left.left = newNode(4);
root.right.left = newNode(5);
root.right.right = newNode(7);
root.left.left.left = newNode(10);
root.left.left.right = newNode(11);
root.right.right.left = newNode(8);
PrintLeafLeftToRight(root);
}
}
// This code is contributed by Rajput-Ji
Javascript
输出:
10 11 5 8
时间复杂度:O(N)
辅助空间: O(N)