先前說了樹的基本操作,我們采用的是二叉鏈表來保存樹形結構,當然二叉有二叉的困擾之處,比如我想找到當前結點的“前驅”和“后繼”,那么我們就必須要遍歷一下樹,然后才能定位到該“節點”的“前驅”和“后繼”,每次定位都是O(n),這不是我們想看到的,那么有什么辦法來解決呢?
從“結構圖”中可以看到,現在結點的指針域中要么是”子節點(SubTree)“或者是”線索(Thread)“,此時就要設立標志位來表示指針域存放的是哪一種。
#region 中序遍歷構建線索二叉樹
/// summary>
/// 中序遍歷構建線索二叉樹
/// /summary>
/// typeparam name="T">/typeparam>
/// param name="tree">/param>
public void BinTreeThreadingCreate_LDRT>(ref ThreadTreeT> tree, ref ThreadTreeT> prevNode)
{
if (tree == null)
return;
//先左子樹遍歷,尋找起始點
BinTreeThreadingCreate_LDR(ref tree.left, ref prevNode);
//如果left為空,則說明該節點可以放“線索”
tree.leftFlag = (tree.left == null) ? NodeFlag.Thread : NodeFlag.SubTree;
//如果right為空,則說明該節點可以放“線索”
tree.rightFlag = (tree.right == null) ? NodeFlag.Thread : NodeFlag.SubTree;
if (prevNode != null)
{
if (tree.leftFlag == NodeFlag.Thread)
tree.left = prevNode;
if (prevNode.rightFlag == NodeFlag.Thread)
prevNode.right = tree;
}
//保存前驅節點
prevNode = tree;
BinTreeThreadingCreate_LDR(ref tree.right, ref prevNode);
}
#endregion
#region 查找指定節點的后繼
/// summary>
/// 查找指定節點的后繼
/// /summary>
/// typeparam name="T">/typeparam>
/// param name="tree">/param>
public ThreadTreeT> BinTreeThreadNext_LDRT>(ThreadTreeT> tree)
{
if (tree == null)
return null;
//如果查找節點的標志域中是Thread,則直接獲取
if (tree.rightFlag == NodeFlag.Thread)
return tree.right;
else
{
//根據中序遍歷的規則是尋找右子樹中中序遍歷的第一個節點
var rightNode = tree.right;
//如果該節點是subTree就需要循環遍歷
while (rightNode.leftFlag == NodeFlag.SubTree)
{
rightNode = rightNode.left;
}
return rightNode;
}
}
#endregion
#region 查找指定節點的前驅
/// summary>
/// 查找指定節點的前驅
/// /summary>
/// typeparam name="T">/typeparam>
/// param name="tree">/param>
/// returns>/returns>
public ThreadTreeT> BinTreeThreadPrev_LDRT>(ThreadTreeT> tree)
{
if (tree == null)
return null;
//如果標志域中存放的是線索,那么可以直接找出來
if (tree.leftFlag == NodeFlag.Thread)
return tree.left;
else
{
//根據”中序“的規則可知,如果不為Thread,則要找出左子樹的最后節點
//也就是左子樹中最后輸出的元素
var leftNode = tree.left;
while (leftNode.rightFlag == NodeFlag.SubTree)
leftNode = leftNode.right;
return leftNode;
}
}
#endregion
#region 遍歷線索二叉樹
/// summary>
/// 遍歷線索二叉樹
/// /summary>
/// typeparam name="T">/typeparam>
/// param name="tree">/param>
public void BinTreeThread_LDRT>(ThreadTreeT> tree)
{
if (tree == null)
return;
while (tree.leftFlag == NodeFlag.SubTree)
tree = tree.left;
do
{
Console.Write(tree.data + "\t");
tree = BinTreeThreadNext_LDR(tree);
} while (tree != null);
}
#endregion
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
namespace ThreadChainTree
{
class Program
{
static void Main(string[] args)
{
ThreadTreeManager manager = new ThreadTreeManager();
//生成根節點
ThreadTreestring> tree = CreateRoot();
//生成節點
AddNode(tree);
ThreadTreestring> prevNode = null;
//構建線索二叉樹
manager.BinTreeThreadingCreate_LDR(ref tree, ref prevNode);
Console.WriteLine("\n線索二叉樹的遍歷結果為:\n");
//中序遍歷線索二叉樹
manager.BinTreeThread_LDR(tree);
}
#region 生成根節點
/// summary>
/// 生成根節點
/// /summary>
/// returns>/returns>
static ThreadTreestring> CreateRoot()
{
ThreadTreestring> tree = new ThreadTreestring>();
Console.WriteLine("請輸入根節點,方便我們生成樹\n");
tree.data = Console.ReadLine();
Console.WriteLine("根節點生成已經生成\n");
return tree;
}
#endregion
#region 插入節點操作
/// summary>
/// 插入節點操作
/// /summary>
/// param name="tree">/param>
static ThreadTreestring> AddNode(ThreadTreestring> tree)
{
ThreadTreeManager mananger = new ThreadTreeManager();
while (true)
{
ThreadTreestring> node = new ThreadTreestring>();
Console.WriteLine("請輸入要插入節點的數據:\n");
node.data = Console.ReadLine();
Console.WriteLine("請輸入要查找的父節點數據:\n");
var parentData = Console.ReadLine();
if (tree == null)
{
Console.WriteLine("未找到您輸入的父節點,請重新輸入。");
continue;
}
Console.WriteLine("請確定要插入到父節點的:1 左側,2 右側");
Direction direction = (Direction)Enum.Parse(typeof(Direction), Console.ReadLine());
tree = mananger.BinTreeThreadAddNode(tree, node, parentData, direction);
Console.WriteLine("插入成功,是否繼續? 1 繼續, 2 退出");
if (int.Parse(Console.ReadLine()) == 1)
continue;
else
break;
}
return tree;
}
#endregion
}
#region 節點標識(用于判斷孩子是節點還是線索)
/// summary>
/// 節點標識(用于判斷孩子是節點還是線索)
/// /summary>
public enum NodeFlag
{
SubTree = 1,
Thread = 2
}
#endregion
#region 線索二叉樹的結構
/// summary>
/// 線索二叉樹的結構
/// /summary>
/// typeparam name="T">/typeparam>
public class ThreadTreeT>
{
public T data;
public ThreadTreeT> left;
public ThreadTreeT> right;
public NodeFlag leftFlag;
public NodeFlag rightFlag;
}
#endregion
#region 插入左節點或者右節點
/// summary>
/// 插入左節點或者右節點
/// /summary>
public enum Direction { Left = 1, Right = 2 }
#endregion
#region 線索二叉樹的基本操作
/// summary>
/// 線索二叉樹的基本操作
/// /summary>
public class ThreadTreeManager
{
#region 將指定節點插入到二叉樹中
/// summary>
/// 將指定節點插入到二叉樹中
/// /summary>
/// typeparam name="T">/typeparam>
/// param name="tree">/param>
/// param name="node">/param>
/// param name="direction">插入做左是右/param>
/// returns>/returns>
public ThreadTreeT> BinTreeThreadAddNodeT>(ThreadTreeT> tree, ThreadTreeT> node, T data, Direction direction)
{
if (tree == null)
return null;
if (tree.data.Equals(data))
{
switch (direction)
{
case Direction.Left:
if (tree.left != null)
throw new Exception("樹的左節點不為空,不能插入");
else
tree.left = node;
break;
case Direction.Right:
if (tree.right != null)
throw new Exception("樹的右節點不為空,不能插入");
else
tree.right = node;
break;
}
}
BinTreeThreadAddNode(tree.left, node, data, direction);
BinTreeThreadAddNode(tree.right, node, data, direction);
return tree;
}
#endregion
#region 中序遍歷構建線索二叉樹
/// summary>
/// 中序遍歷構建線索二叉樹
/// /summary>
/// typeparam name="T">/typeparam>
/// param name="tree">/param>
public void BinTreeThreadingCreate_LDRT>(ref ThreadTreeT> tree, ref ThreadTreeT> prevNode)
{
if (tree == null)
return;
//先左子樹遍歷,尋找起始點
BinTreeThreadingCreate_LDR(ref tree.left, ref prevNode);
//如果left為空,則說明該節點可以放“線索”
tree.leftFlag = (tree.left == null) ? NodeFlag.Thread : NodeFlag.SubTree;
//如果right為空,則說明該節點可以放“線索”
tree.rightFlag = (tree.right == null) ? NodeFlag.Thread : NodeFlag.SubTree;
if (prevNode != null)
{
if (tree.leftFlag == NodeFlag.Thread)
tree.left = prevNode;
if (prevNode.rightFlag == NodeFlag.Thread)
prevNode.right = tree;
}
//保存前驅節點
prevNode = tree;
BinTreeThreadingCreate_LDR(ref tree.right, ref prevNode);
}
#endregion
#region 查找指定節點的后繼
/// summary>
/// 查找指定節點的后繼
/// /summary>
/// typeparam name="T">/typeparam>
/// param name="tree">/param>
public ThreadTreeT> BinTreeThreadNext_LDRT>(ThreadTreeT> tree)
{
if (tree == null)
return null;
//如果查找節點的標志域中是Thread,則直接獲取
if (tree.rightFlag == NodeFlag.Thread)
return tree.right;
else
{
//根據中序遍歷的規則是尋找右子樹中中序遍歷的第一個節點
var rightNode = tree.right;
//如果該節點是subTree就需要循環遍歷
while (rightNode.leftFlag == NodeFlag.SubTree)
{
rightNode = rightNode.left;
}
return rightNode;
}
}
#endregion
#region 查找指定節點的前驅
/// summary>
/// 查找指定節點的前驅
/// /summary>
/// typeparam name="T">/typeparam>
/// param name="tree">/param>
/// returns>/returns>
public ThreadTreeT> BinTreeThreadPrev_LDRT>(ThreadTreeT> tree)
{
if (tree == null)
return null;
//如果標志域中存放的是線索,那么可以直接找出來
if (tree.leftFlag == NodeFlag.Thread)
return tree.left;
else
{
//根據”中序“的規則可知,如果不為Thread,則要找出左子樹的最后節點
//也就是左子樹中最后輸出的元素
var leftNode = tree.left;
while (leftNode.rightFlag == NodeFlag.SubTree)
leftNode = leftNode.right;
return leftNode;
}
}
#endregion
#region 遍歷線索二叉樹
/// summary>
/// 遍歷線索二叉樹
/// /summary>
/// typeparam name="T">/typeparam>
/// param name="tree">/param>
public void BinTreeThread_LDRT>(ThreadTreeT> tree)
{
if (tree == null)
return;
while (tree.leftFlag == NodeFlag.SubTree)
tree = tree.left;
do
{
Console.Write(tree.data + "\t");
tree = BinTreeThreadNext_LDR(tree);
} while (tree != null);
}
#endregion
}
#endregion
}