Mirror image of a binary tree
Suppose there is a tree:
1
/ \
2 3
/ \
4 5
Then the mirror image will be:
开发者_运维百科 1
/ \
3 2
/ \
5 4
Assume the nodes are of this structure:
struct node{
node left;
node right;
int value;
}
Can someone suggest an algorithm for this?
Sounds like homework.
It looks very easy. Write a recursive routine that depth-first visits every node and builds the mirror tree with left and right reversed.
struct node *mirror(struct node *here) {
if (here == NULL)
return NULL;
else {
struct node *newNode = malloc (sizeof(struct node));
newNode->value = here->value;
newNode->left = mirror(here->right);
newNode->right = mirror(here->left);
return newNode;
}
}
This returns a new tree - some other answers do this in place. Depends on what your assignment asked you to do :)
void swap_node(node n) {
if(n != null) {
node tmp = n.left;
n.left = n.right;
n.right = tmp;
swap_node(n.left);
swap_node(n.right);
}
}
swap_node(root);
Banal solution:
for each node in tree
exchange leftchild with rightchild.
Recursive and Iterative methods in JAVA: 1) Recursive:
public static TreeNode mirrorBinaryTree(TreeNode root){
if(root == null || (root.left == null && root.right == null))
return root;
TreeNode temp = root.left;
root.left = root.right;
root.right = temp;
mirrorBinaryTree(root.left);
mirrorBinaryTree(root.right);
return root;
}
2) Iterative:
public static TreeNode mirrorBinaryTreeIterative(TreeNode root){
if(root == null || (root.left == null && root.right == null))
return root;
TreeNode parent = root;
Stack<TreeNode> treeStack = new Stack<TreeNode>();
treeStack.push(root);
while(!treeStack.empty()){
parent = treeStack.pop();
TreeNode temp = parent.right;
parent.right = parent.left;
parent.left = temp;
if(parent.right != null)
treeStack.push(parent.right);
if(parent.left != null)
treeStack.push(parent.left);
}
return root;
}
void mirror(struct node* node)
{
if (node==NULL)
{
return;
}
else
{
struct node* temp;
mirror(node->left);
mirror(node->right);
temp = node->left;
node->left = node->right;
node->right = temp;
}
}
An iterative solution:
public void mirrorIterative() {
Queue<TreeNode> nodeQ = new LinkedList<TreeNode>();
nodeQ.add(root);
while(!nodeQ.isEmpty()) {
TreeNode node = nodeQ.remove();
if(node.leftChild == null && node.rightChild == null)
continue;
if(node.leftChild != null && node.rightChild != null) {
TreeNode temp = node.leftChild;
node.leftChild = node.rightChild;
node.rightChild = temp;
nodeQ.add(node.leftChild);
nodeQ.add(node.rightChild);
}
else if(node.leftChild == null) {
node.leftChild = node.rightChild;
node.rightChild = null;
nodeQ.add(node.leftChild);
} else {
node.rightChild = node.leftChild;
node.leftChild = null;
nodeQ.add(node.rightChild);
}
}
}
void mirror(node<t> *& root2,node<t> * root)
{
if(root==NULL)
{
root2=NULL;
}
else {
root2=new node<t>;
root2->data=root->data;
root2->left=NULL;
root2->right=NULL;
mirror(root2->left,root->right);
mirror(root2->right,root->left);
}
}
TreeNode * mirror(TreeNode *node){
if(node==NULL){
return NULL;
}else{
TreeNode *temp=node->left;
node->left=mirror(node->right);
node->right=mirror(temp);
return node;
}
}
Here is my function. Do suggest if any better solution:
void mirrorimage(struct node *p)
{
struct node *q;
if(p!=NULL)
{
q=swaptrs(&p);
p=q;
mirrorimage(p->left);
mirrorimage(p->right);
}
}
struct node* swaptrs(struct node **p)
{
struct node *temp;
temp=(*p)->left;
(*p)->left=(*p)->right;
(*p)->right=temp;
return (*p);
}
Recursive Java Code
public class TreeMirrorImageCreator {
public static Node createMirrorImage(Node originalNode,Node mirroredNode){
mirroredNode.setValue(originalNode.getValue());
if(originalNode.getLeft() != null){
mirroredNode.setLeft(createMirrorImage(originalNode.getRight(),new Node(0)));
}
if(originalNode.getRight() != null){
mirroredNode.setRight(createMirrorImage(originalNode.getLeft(), new Node(0)));
}
return mirroredNode;
}
}
struct node *MirrorOfBinaryTree( struct node *root)
{ struct node *temp;
if(root)
{
MirrorOfBinaryTree(root->left);
MirrorOfBinaryTree(root->right);
/*swap the pointers in this node*/
temp=root->right;
root->right=root->left;;
root->left=temp;
}
return root;
}
Time complexity: O(n) Space complexity: O(n)
Well, this ques has got a lot of answers.I am posting an iterative version for this which is quite easy to understand .This uses level order Traversal.
//Function for creating Binary Tree
//Assuming *root for original tree and *root2 for mirror tree
//are declared globally
void insert(int data)
{
struct treenode* temp;
if(root2 == NULL)
{
root2 = createNode(data);
return;
}
else{
queue<treenode*> q;
q.push(root2);
while(!q.empty())
{
temp = q.front();
q.pop();
if(temp->left)
q.push(temp->left);
else{
temp->left = createNode(data);
return;
}
if(temp->right)
{
q.push(temp->right);
}
else{
temp -> right = createNode(data);
return;
}
}
}
}
//Iterative version for Mirror of a Binary Tree
void mirrorBinaryTree()
{
struct treenode *front;
queue<treenode*> q;
q.push(root);
while(!q.empty())
{
front = q.front();
insert(front->data);
q.pop();
if(front->right)
q.push(front->right);
if(front->left)
q.push(front->left);
}
}
Here's a non-recursive way of doing it in python using queues. The Queue class may be initialised like this:
class Queue(object):
def __init__(self):
self.items = []
def enqueue(self, item):
self.items.insert(0, item)
def dequeue(self):
if not self.is_empty():
return self.items.pop()
def is_empty(self):
return len(self.items) == 0
def peek(self):
if not self.is_empty():
return self.items[-1]
def __len__(self):
return self.size()
def size(self):
return len(self.items)
The class which represents a node:
class Node:
def __init__(self, data):
self.left = None
self.right = None
self.val = data
And here's the method along with traversal example which does the task:
class BinaryTree(object):
def __init__(self, root):
self.root = Node(root)
def inorder(self, start, traversal):
if start:
traversal = self.inorder(start.left, traversal)
traversal += f"{start.val} "
traversal = self.inorder(start.right, traversal)
return traversal
def mirror_tree_iterative(root):
if root is None:
return
q = Queue()
q.enqueue(root)
while not q.is_empty():
curr = q.peek()
q.dequeue()
curr.left, curr.right = curr.right, curr.left
if curr.left:
q.enqueue(curr.left)
if curr.right:
q.enqueue(curr.right)
tree = BinaryTree(1)
tree.root.left = Node(2)
tree.root.right = Node(3)
tree.root.left.left = Node(4)
tree.root.left.right = Node(5)
tree.root.right.left = Node(6)
print(tree.inorder(tree.root, ''))
mirror_tree_iterative(tree.root)
print(tree.inorder(tree.root, ''))
精彩评论