给定二叉树,任务是按树的“层”顺序遍历的顺序对其进行展平。在扁平化的二叉树中,所有节点的左节点必须为NULL。
例子:
Input:
1
/ \
5 2
/ \ / \
6 4 9 3
Output: 1 5 2 6 4 9 3
Input:
1
\
2
\
3
\
4
\
5
Output: 1 2 3 4 5
方法:我们将通过模拟二叉树的Level顺序遍历来解决此问题,如下所示:
- 创建一个队列来存储二叉树的节点。
- 创建一个变量“ prev”,并通过父节点对其进行初始化。
- 将左右父母的孩子推入队列。
- 应用级别订单遍历。可以说“ curr”是队列中最靠前的元素。然后,
- 如果’curr’为NULL,则继续。
- 否则将curr-> left和curr-> right推送到队列中
- 设置上一个= curr
下面是上述方法的实现:
C++
// C++ implementation of the approach
#include
using namespace std;
// Node of the Binary tree
struct node {
int data;
node* left;
node* right;
node(int data)
{
this->data = data;
left = NULL;
right = NULL;
}
};
// Function to flatten Binary tree using
// level order traversal
void flatten(node* parent)
{
// Queue to store nodes
// for BFS
queue q;
q.push(parent->left);
q.push(parent->right);
node* prev = parent;
// Code for BFS
while (q.size()) {
// Size of queue
int s = q.size();
while (s--) {
// Front most node in
// the queue
node* curr = q.front();
q.pop();
// Base case
if (curr == NULL)
continue;
prev->right = curr;
prev->left = NULL;
prev = curr;
// Pushing new elements
// in queue
q.push(curr->left);
q.push(curr->right);
}
}
prev->left = NULL;
prev->right = NULL;
}
// Function to print flattened
// Binary Tree
void print(node* parent)
{
node* curr = parent;
while (curr != NULL)
cout << curr->data << " ", curr = curr->right;
}
// Driver code
int main()
{
node* root = new node(1);
root->left = new node(5);
root->right = new node(2);
root->left->left = new node(6);
root->left->right = new node(4);
root->right->left = new node(9);
root->right->right = new node(3);
// Calling required functions
flatten(root);
print(root);
return 0;
}
Java
// Java implementation of the approach
import java.util.*;
class GFG
{
// Node of the Binary tree
static class node
{
int data;
node left;
node right;
node(int data)
{
this.data = data;
left = null;
right = null;
}
};
// Function to flatten Binary tree using
// level order traversal
static void flatten(node parent)
{
// Queue to store nodes
// for BFS
Queue q = new LinkedList<>();
q.add(parent.left);
q.add(parent.right);
node prev = parent;
// Code for BFS
while (q.size() > 0)
{
// Size of queue
int s = q.size();
while (s-- > 0)
{
// Front most node in
// the queue
node curr = q.peek();
q.remove();
// Base case
if (curr == null)
continue;
prev.right = curr;
prev.left = null;
prev = curr;
// Pushing new elements
// in queue
q.add(curr.left);
q.add(curr.right);
}
}
prev.left = null;
prev.right = null;
}
// Function to print flattened
// Binary Tree
static void print(node parent)
{
node curr = parent;
while (curr != null)
{
System.out.print(curr.data + " ");
curr = curr.right;
}
}
// Driver code
public static void main(String[] args)
{
node root = new node(1);
root.left = new node(5);
root.right = new node(2);
root.left.left = new node(6);
root.left.right = new node(4);
root.right.left = new node(9);
root.right.right = new node(3);
// Calling required functions
flatten(root);
print(root);
}
}
// This code is contributed by Rajput-Ji
Python
# Python implementation of above algorithm
# Utility class to create a node
class node:
def __init__(self, key):
self.data = key
self.left = self.right = None
# Function to flatten Binary tree using
# level order traversal
def flatten( parent):
# Queue to store nodes
# for BFS
q = []
q.append(parent.left)
q.append(parent.right)
prev = parent
# Code for BFS
while (len(q) > 0) :
# Size of queue
s = len(q)
while (s > 0) :
s = s - 1
# Front most node in
# the queue
curr = q[0]
q.pop(0)
# Base case
if (curr == None):
continue
prev.right = curr
prev.left = None
prev = curr
# appending elements
# in queue
q.append(curr.left)
q.append(curr.right)
prev.left = None
prev.right = None
# Function to print flattened
# Binary Tree
def print_(parent):
curr = parent
while (curr != None):
print( curr.data , end=" ")
curr = curr.right
# Driver code
root = node(1)
root.left = node(5)
root.right = node(2)
root.left.left = node(6)
root.left.right = node(4)
root.right.left = node(9)
root.right.right = node(3)
# Calling required functions
flatten(root)
print_(root)
# This code is contributed by Arnab Kundu
C#
// C# implementation of the approach
using System;
using System.Collections.Generic;
class GFG
{
// Node of the Binary tree
public class node
{
public int data;
public node left;
public node right;
public node(int data)
{
this.data = data;
left = null;
right = null;
}
};
// Function to flatten Binary tree using
// level order traversal
static void flatten(node parent)
{
// Queue to store nodes
// for BFS
Queue q = new Queue();
q.Enqueue(parent.left);
q.Enqueue(parent.right);
node prev = parent;
// Code for BFS
while (q.Count > 0)
{
// Size of queue
int s = q.Count;
while (s-- > 0)
{
// Front most node in
// the queue
node curr = q.Peek();
q.Dequeue();
// Base case
if (curr == null)
continue;
prev.right = curr;
prev.left = null;
prev = curr;
// Pushing new elements
// in queue
q.Enqueue(curr.left);
q.Enqueue(curr.right);
}
}
prev.left = null;
prev.right = null;
}
// Function to print flattened
// Binary Tree
static void print(node parent)
{
node curr = parent;
while (curr != null)
{
Console.Write(curr.data + " ");
curr = curr.right;
}
}
// Driver code
public static void Main(String[] args)
{
node root = new node(1);
root.left = new node(5);
root.right = new node(2);
root.left.left = new node(6);
root.left.right = new node(4);
root.right.left = new node(9);
root.right.right = new node(3);
// Calling required functions
flatten(root);
print(root);
}
}
// This code is contributed by Rajput-Ji
输出 :
1 5 2 6 4 9 3
时间复杂度: O(N)
空间复杂度: O(N),其中N是二叉树的大小。