重拾算法(2)——线索二叉树

重拾算法(2)——线索二叉树

上一篇我们实现了二叉树的递归和非递归遍历，并为其复用精心设计了遍历方法Traverse(TraverseOrder order, NodeWorker<T> worker);今天就在此基础上实现线索二叉树。

什么是线索二叉树

二叉树中容易找到结点的左右孩子信息，但该结点在某一序列中的直接前驱和直接后继只能在某种遍历过程中动态获得。

先依遍历规则把每个结点某一序列中对应的前驱和后继线索预存起来，这叫做"线索化"。

意义：从任一结点出发都能快速找到其某一序列中前驱和后继，且不必借助堆栈。

这就是线索二叉树（Threaded Binary Tree）

数据结构

如何预存这类信息？有两种解决方法。

每个结点增加两个域：fwd和bwd

与原有的左右孩子指针域"复用"

任意一个二叉树，其结点数为N，则有N+1个空链域。空链域就是叶结点的lchild和rchild。比如下面这个二叉树

它有7个结点，有（7+1）个空链域。这个结论可以用数学归纳法证明。

大体思路就是充分利用那n+1个空链域，用它存储前驱结点和后继结点。

我们规定：

• 若结点有左子树，则lchild指向其左孩子；否则，lchild指向其直接前驱(即线索)；
• 若结点有右子树，则rchild指向其右孩子；否则，rchild指向其直接后继(即线索) 。

那么就有如下一个问题：如何判断是孩子指针还是线索指针？

容易的很，加两个标志域：

标志域只需要2个bit，而用增加指针域的方式则需要2个指针的空间（一个指针就是一个int的长度）。所以这个方案极大地节省了空间。

我们规定：

• 当Tag域为0时,表示孩子情况；
• 当Tag域为1时,表示线索情况。

有关线索二叉树的几个术语

线索链表：用含Tag的结点样式所构成的二叉链表。

线 索：指向结点前驱和后继的指针。

线索二叉树：加上线索的二叉树。

线 索 化：对二叉树以某种次序遍历使其变为线索二叉树的过程。

线索化过程就是在遍历过程中修改空指针的过程：

• 将空的lchild改为结点的直接前驱；
• 将空的rchild改为结点的直接后继；
• 非空指针呢？仍然指向孩子结点。（称为"正常情况"）

C#实现线索二叉树

数据结构

线索二叉树是（is a）二叉树。所以最初我想继承上一篇的二叉树BinaryTreeNode<T>类型。但后来发现代码不太好掌控，不如来得直接些，让二叉树和线索二叉树作为两个毫不相干的类型。以后重构的时候再考虑考虑如何整合。

 

    public partial class ThreadedBinaryTreeNode<T>
    {
        public T Value { get;set; }
        public ThreadedBinaryTreeNode<T> Parent { get;set; }
        public ThreadedBinaryTreeNode<T> Left { get;set; }
        public ThreadedBinaryTreeNode<T> Right { get;set; }
        public bool Point2PreviousNode { get; private set; }
        public bool Point2NextNode { get; private set; }
        public TraverseOrder Order { get;set; }
        private bool threadedOnce = false;
        
        public ThreadedBinaryTreeNode(T value, ThreadedBinaryTreeNode<T> parent = null, ThreadedBinaryTreeNode<T> left = null, ThreadedBinaryTreeNode<T> right = null)
        {
            this.Value = value; this.Parent = parent; this.Left = left; this.Right = right;
        }

        public void Traverse(TraverseOrder order, ThreadedNodeWorker<T> worker)
        {
            if (worker == null) { return; }

            if ((!threadedOnce) || (order != this.Order))
            {
                var router = new Router<T>(worker);
                this.NormalTraverse(order, router);
                this.Order = order;
                this.threadedOnce = true;
            }
            else
            {
                this.ThreadedTraverse(order, worker);
            }
        }
    }
    
    public abstract class ThreadedNodeWorker<T>
    {
        public abstract void DoActionOnNode(ThreadedBinaryTreeNode<T> node);
    }

C#里用bool类型作为标志变量大概再好不过了。

线索化

线索二叉树的线索化是首先要解决的问题。

    public partial class ThreadedBinaryTreeNode<T>
    {
        /* 略 */
        private void RefreshThread(TraverseOrder order)
        {
            var threader = new Threader<T>();
            this.NormalTraverse (order, threader);
            this.Order = order;
        }
        
        class Threader<TThreader> : ThreadedNodeWorker<TThreader>
        {
            Queue<ThreadedBinaryTreeNode<TThreader>> queue = new Queue<ThreadedBinaryTreeNode<TThreader>>();
            public override void DoActionOnNode(ThreadedBinaryTreeNode<TThreader> node)
            {
                this.queue.Enqueue(node);
                if (this.queue.Count <= 1) { return; }
                
                var pre = this.queue.Dequeue();
                var next = this.queue.Peek();
                if (pre.Right == null)
                {
                    pre.Right = next;
                    pre.Point2NextNode = true;
                }
                if (next.Left == null)
                {
                    next.Left = pre;
                    next.Point2PreviousNode = true;
                }
            }
        }
    }

代码中的this.NormalTraverse()函数与上一篇中二叉树的遍历方法Traverse()思路是一样的，唯一区别在于传入的操作结点的对象threader，这个threader在用前序、中序、后序或层次遍历二叉树时，将结点存入一个队列queue，当queue中有2个结点时，就设置他们互为前驱后驱（若此2个结点的Left/Right为空的前提下）。简单来说，就是遍历二叉树，根据访问结点的先后顺序依次修改其前驱后驱指针（及标识）

完成线索化后，就可以遍历了。

前序遍历

前序遍历线索二叉树的思路是：对于一个结点A及其左右子树Left和Right，先访问A；如果A的标记Point2NextNode为true，则A的下一结点通过A.Right就能确定；否则，有两种情况：(1)A的下一结点在A.Left上，而以A.Left为根的这个子树的第一个被访问的结点，恰恰是A.Left这个结点。(2)A.Left为空，这说明A的下一结点在A.Right上。若A.Right也为空，就说明整个树遍历结束。

        void ThreadedPreorderTraverse(ThreadedNodeWorker<T> worker)
        {
            var node = this;
            while (node != null)
            {
                var p2next = node.Point2NextNode;
                var p2pre = node.Point2PreviousNode;
                var right = node.Right;
                var left = node.Left;
                worker.DoActionOnNode(node);
                if (p2next)
                {
                    node = right;
                }
                else
                {
                    if (p2pre)
                    {
                        node = right;
                    }
                    else
                    {
                        if (left != null)
                        { node = left; }
                        else
                        { node = right; }
                    }
                }
            }
        }

中序遍历

中序遍历线索二叉树的思路是：对于一个结点A及其左右子树Left和Right，第一个要访问的结点在A.Left子树，A.Left子树中第一个要访问的结点则在A.Left.Left子树中，依次循环，直到最后一个叶结点的Left。所以A树要访问的第一个结点就用GetLeftMostNode()这个函数去找。访问完A，就该A的右子树了，所以A.Right就成为下一个A了，所以就用GetLeftMostNode(A.Right)获取下一个结点。

        ThreadedBinaryTreeNode<T> GetLeftMostNode(ThreadedBinaryTreeNode<T> node)
        {
            if (node == null) { return null; }
            while ((node.Left != null) && (!node.Point2PreviousNode))
            {
                node = node.Left;
            }
            return node;
        }

        void ThreadedInorderTraverse(ThreadedNodeWorker<T> worker)
        {
            var node = GetLeftMostNode(this);
            while (node != null)
            {
                var p2next = node.Point2NextNode;
                var p2pre = node.Point2PreviousNode;
                var right = node.Right;
                var left = node.Left;
                worker.DoActionOnNode(node);
                if (p2next)
                {
                    node = right;
                }
                else
                {
                    node = GetLeftMostNode(right);// visit right sub tree.
                }
            }
        }

后续遍历

后序遍历线索二叉树的思路是：对于一个结点A及其左右子树Left和Right，第一个要访问的结点在A.Left子树，A.Left子树中第一个要访问的结点则在A.Left.Left子树中，依次循环，直到最后一个叶结点的Left。所以A树要访问的第一个结点就用GetLeftMostNode()这个函数去找。访问完A树，下一步分三种情况：(1)A是整个树的根，这时整个遍历结束。(2)A树是A.Parent的左子树，这时就该A.Parent.Right了，于是A.Parent.Right就成了之前的A。(3)A树是A.Parent的右子树，这时就该A结点了。

        ThreadedBinaryTreeNode<T> GetFirstNode4PostorderTraverse(ThreadedBinaryTreeNode<T> root)
        {
            if (root == null) { return null; }
            
            ThreadedBinaryTreeNode<T> result = null;
            var stack = new Stack<ThreadedBinaryTreeNode<T>>(); var stackReady4Visit = new Stack<bool>();
            
            stack.Push(root); stackReady4Visit.Push(false);
            
            while (stack.Count > 0)
            {
                var node = stack.Pop(); var ready4Visit = stackReady4Visit.Pop();
                //if (node == null) { continue; }
                if (ready4Visit)
                {
                    result = node;
                    break;
                }
                else
                {
                    var right = node.Right;
                    var left = node.Left;
                    var p2pre = node.Point2PreviousNode;
                    var p2next = node.Point2NextNode;
                    stack.Push(node); stackReady4Visit.Push(true);
                    if ((right != null) && (!p2next)) { stack.Push(right); stackReady4Visit.Push(false); }
                    if ((left != null) && (!p2pre)) { stack.Push(left); stackReady4Visit.Push(false); }
                }
            }
            return result;
        }
        
        void ThreadedPostorderTraverse(ThreadedNodeWorker<T> worker)
        {
            var node = GetFirstNode4PostorderTraverse(this);
            
            while (node != null)
            {
                var p2next = node.Point2NextNode;
                var p2pre = node.Point2PreviousNode;
                var right = node.Right;
                var left = node.Left;
                worker.DoActionOnNode(node);
                if (p2next)
                {
                    node = right;
                }
                else
                {
                    var parent = node.Parent;
                    if (parent != null)// node is NOT root
                    {
                        if (parent.Right == node)// node is right sub tree of parent
                        {
                            node = parent;
                        }
                        else // node is left sub tree of parent
                        {
                            if (parent.Point2NextNode)
                            {
                                node = parent;
                            }
                            else
                            {
                                node = GetFirstNode4PostorderTraverse(parent.Right);
                                if (node == null)
                                {
                                    node = parent;
                                }
                            }
                        }
                    }
                    else// node is root of whole tree
                    {
                        node = parent;
                    }
                }
            }
        }

层次遍历

线索二叉树对于层次遍历似乎没有优化的能力，其层次遍历仍旧用普通二叉树的遍历方法好了。

        void NormalLayerTraverse(ThreadedNodeWorker<T> worker)
        {
            var queue = new Queue<ThreadedBinaryTreeNode<T>>();
            queue.Enqueue(this);
            while (queue.Count > 0)
            {
                var node = queue.Dequeue();
                if (node != null)
                {
                    var left = node.Left;
                    var right = node.Right;
                    var p2pre = node.Point2PreviousNode;
                    var p2next = node.Point2NextNode;
                    worker.DoActionOnNode(node);
                    if ((left != null) && (!p2pre)) { queue.Enqueue(left); }
                    if ((right != null) && (!p2next)) { queue.Enqueue(right); }
                }
            }
        }

总结

学习了线索二叉树后，我对算法思维有了新的认识。在想一个子问题的时候，以不同的角度思考，得到不同的结论，将其连接起来，竟然能够得到整个问题的解。