您的位置:首页 > 其它

算法_二叉树反转

2016-11-22 16:15 148 查看
前一阵homebrew作者面试谷歌被拒,原因之一是这位老兄无法反转出二叉树。

 既然众公司面试都爱用这货面试,咱也来做一下。 先定义二叉树类
public class BinaryTreeNode<T>
{
public string Name { get; set; }
public T Data { get; set; }
public BinaryTreeNode<T> Left { get; set; }
public BinaryTreeNode<T> Right { get; set; }
private bool _isLeaf;
public bool IsLeaf
{
get
{
_isLeaf = (Left == null && Right == null);
return _isLeaf;
}
}
public BinaryTreeNode(string name, T data = default(T))
{
Name = name;
Data = data;
}
public BinaryTreeNode<T> CreateAndJionLeft(string name, T data = default(T))
{
return Left = new BinaryTreeNode<T>(name, data);
}
public BinaryTreeNode<T> CreateAndJionRight(string name, T data = default(T))
{
return Right = new BinaryTreeNode<T>(name, data);
}
}

Name和Data是二叉树内部元素,根据需求调整即可,CreateAndJionLeft表示将左边子节点加入当前节点,也可在外部调用端实例化子节点。 接下来是二叉树常用的几种遍历方式:前序遍历,中序遍历,后序遍历,逐层遍历
/// <summary>
/// 前序遍历
/// </summary>
public void PreOrderTraversal()
{
if (this != null)
{
Trace.WriteLine(Name);
}
if (Left != null)
{
Left.PreOrderTraversal();
}
if (Right != null)
{
Right.PreOrderTraversal();
}
}
/// <summary>
/// 前序遍历 非递归
/// </summary>
public void PreOrderTraversalEx()
{
Stack<BinaryTreeNode<T>> stack = new Stack<BinaryTreeNode<T>>();
BinaryTreeNode<T> node = this;
while (node != null || stack.Count != 0)
{
while (node != null)
{
Trace.WriteLine(node.Name);
stack.Push(node);
node = node.Left;
}
if (stack.Count != 0)
{
node = stack.Pop();
node = node.Right;
}
}
}

/// <summary>
/// 中序遍历
/// </summary>
public void InOrderTraversal()
{
if (Left != null)
{
Left.InOrderTraversal();
}
if (this != null)
{
Trace.WriteLine(Name);
}
if (Right != null)
{
Right.InOrderTraversal();
}
}
/// <summary>
/// 中序遍历 非递归
/// </summary>
public void InOrderTraversalEx()
{
Stack<BinaryTreeNode<T>> stack = new Stack<BinaryTreeNode<T>>();
BinaryTreeNode<T> node = this;
while (node != null || stack.Count != 0)
{
while (node != null)
{
stack.Push(node);
node = node.Left;
}
if (stack.Count != 0)
{
node = stack.Peek();
stack.Pop();
Trace.WriteLine(node.Name);
node = node.Right;
}
}
}

/// <summary>
/// 后序遍历
/// </summary>
public void PostOrderTraversal()
{
if (Left != null)
{
Left.PostOrderTraversal();
}
if (Right != null)
{
Right.PostOrderTraversal();
}
if (this != null)
{
Trace.WriteLine(Name);
}
}
/// <summary>
/// 后序遍历 非递归
/// </summary>
public void PostOrderTraversalEx()
{
Stack<BinaryTreeNode<T>> stack = new Stack<BinaryTreeNode<T>>();
BinaryTreeNode<T> node = this;
BinaryTreeNode<T> lastNode = null;
stack.Push(node);
while (stack.Count!=0)
{
node = stack.Peek();
if ((node.Left == null && node.Right == null) || (lastNode != null && (lastNode == node.Left || lastNode == node.Right)))
{
//如果当前结点没有子结点或者子节点都已被访问过就输出  因为后序遍历是先子节点再根节点
Trace.WriteLine(node.Name);
stack.Pop();
lastNode = node;//设定本次访问的节点为上一次访问的节点
}
else
{
if (node.Right != null)
stack.Push(node.Right);
if (node.Left != null)
stack.Push(node.Left);
}
}
}

/// <summary>
/// 逐层遍历
/// </summary>
public void Traversal()
{
Stack<BinaryTreeNode<T>> stack = new Stack<BinaryTreeNode<T>>();
stack.Push(this);
while (stack.Count > 0)
{
BinaryTreeNode<T> temp = stack.Peek();
Trace.WriteLine(temp.Name);
stack.Pop();
if (temp.Left != null)
{
stack.Push(temp.Left);
}
if (temp.Right != null)
{
stack.Push(temp.Right);
}
}
}

 递归实现反转二叉树
/// <summary>
/// 反转二叉树(递归)
/// </summary>
public BinaryTreeNode<T> ReverseWithRecursive()
{
if (this == null)
{
return null;
}

if (!(Left == null && Right == null))
{
BinaryTreeNode<T> temp = Right;//左右节点反转
Right = Left;
Left = temp;

if (Left != null)
Left = Left.ReverseWithRecursive();//递归反转左子节点
if (Right != null)
Right = Right.ReverseWithRecursive();//递归反转右子节点
}
return this;
}

 非递归方式实现反转二叉树
     /// <summary>
/// 反转二叉树(非递归)
/// </summary>
/// <returns></returns>
public BinaryTreeNode<T> Reverse()
{
if (this == null)
{
return null;
}

Queue<BinaryTreeNode<T>> nodeQueue = new Queue<BinaryTreeNode<T>>();
nodeQueue.Enqueue(this);
while (nodeQueue.Count > 0)
{
BinaryTreeNode<T> node = nodeQueue.Peek();
nodeQueue.Dequeue();

BinaryTreeNode<T> tempNode = node.Right;//左右节点反转
node.Right = node.Left;
node.Left = tempNode;

if (node.Left != null)
{
nodeQueue.Enqueue(node.Left);
}
if (node.Right != null)
{
nodeQueue.Enqueue(node.Right);
}
}
return this;
}

定义一个队列来临时存储即将反转的节点。获取队列中存储的节点,获取到一个节点后队列中的此节点已无用将其删除,然后把获取到的节点的左右子节点反转,将反转后的左右子节点都放入队列用于下一次循环。重复执行直到反转完所有树节点。 换成栈存储
/// <summary>
/// 反转二叉树(非递归) 栈
/// </summary>
/// <returns></returns>
public BinaryTreeNode<T> ReverseEx()
{
if (this == null)
{
return null;
}

Stack<BinaryTreeNode<T>> nodeStack = new Stack<BinaryTreeNode<T>>();
nodeStack.Push(this);
while (nodeStack.Count > 0)
{
BinaryTreeNode<T> node = nodeStack.Peek();
nodeStack.Pop();

BinaryTreeNode<T> tempNode = node.Right;//左右节点反转
node.Right = node.Left;
node.Left = tempNode;

if (node.Left != null)
{
nodeStack.Push(node.Left);
}
if (node.Right != null)
{
nodeStack.Push(node.Right);
}
}
return this;
}

  客户端调用
//创建二叉树
BinaryTreeNode<int> tree = new BinaryTreeNode<int>("0");//rootNode
BinaryTreeNode<int> n01 = tree.CreateAndJionLeft("01");
BinaryTreeNode<int> n02 = tree.CreateAndJionRight("02");
BinaryTreeNode<int> n011 = n01.CreateAndJionLeft("011");
BinaryTreeNode<int> n012 = n01.CreateAndJionRight("012");
BinaryTreeNode<int> n021 = n02.CreateAndJionLeft("021");
BinaryTreeNode<int> n0211 = n021.CreateAndJionLeft("0211");
BinaryTreeNode<int> n0212 = n021.CreateAndJionRight("0212");

//遍历输出
tree.Traversal();

//反转二叉树并遍历输出
BinaryTreeNode<int> treeReverse = tree.Reverse();
treeReverse.Traversal();

反转前

 反转后

 转载请注明地址:http://www.cnblogs.com/wintersoft/p/4676124.html
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  二叉树 反转 非递归