package com.testtree;

/**
 * @author pier
 */
public class TreeNode {
    /**數(shù)據(jù)域**/
    private int data;
    /**左指針**/
    private TreeNode left;
    /** 左孩子是否為線索，采用布爾類型主要是判斷是否未null足以**/
    private boolean leftIsThread;
    /**右指針**/
    private TreeNode right;
    /** 右孩子是否為線索**/
    private boolean rightIsThread;

    /**根據(jù)數(shù)據(jù)域來確定所在的指針對(duì)應(yīng)位置**/
    public TreeNode(int data)
    {
        this.data = data;
        this.left = null;
        this.leftIsThread = false;
        this.right = null;
        this.rightIsThread = false;
    }

    public int getData()
    {
        return data;
    }

    public void setData(int data)
    {
        this.data = data;
    }

    public TreeNode getLeft()
    {
        return left;
    }

    public void setLeft(TreeNode left)
    {
        this.left = left;
    }

    public boolean isLeftIsThread()
    {
        return leftIsThread;
    }

    public void setLeftIsThread(boolean leftIsThread)
    {
        this.leftIsThread = leftIsThread;
    }

    public TreeNode getRight()
    {
        return right;
    }

    public void setRight(TreeNode right)
    {
        this.right = right;
    }

    public boolean isRightIsThread()
    {
        return rightIsThread;
    }

    public void setRightIsThread(boolean rightIsThread)
    {
        this.rightIsThread = rightIsThread;
    }

    @Override
    public boolean equals(Object obj)
    {
        if (obj instanceof TreeNode )
        {
            TreeNode temp = (TreeNode) obj;
            if (temp.getData() == this.data)
            {
                return true;
            }
        }
        return false;
    }

    @Override
    public int hashCode()
    {
        return super.hashCode() + this.data;
    }
}

package com.testtree;
/*author:pier
2021/10/12
*/

public class BiTree {
    /** 根節(jié)點(diǎn) **/
    private TreeNode root;
    /** 大小 **/
    private int size;
    /** 線索化的時(shí)候保存前驅(qū) **/
    private TreeNode pre = null;

    public BiTree()
    {
        this.root = null;
        this.size = 0;
        this.pre = null;
    }

    public BiTree(int[] data)
    {
        this.pre = null;
        this.size = data.length;
        // 創(chuàng)建二叉樹
        this.root = createTree(data, 1);
    }

    /**
     * 創(chuàng)建二叉樹
     *
     */
    public TreeNode createTree(int[] data, int index)
    {
        if (index > data.length)
        {
            return null;
        }
        TreeNode node = new TreeNode(data[index - 1]);
        TreeNode left = createTree(data, 2 * index);
        TreeNode right = createTree(data, 2 * index + 1);
        node.setLeft(left);
        node.setRight(right);
        return node;
    }
    /**中序遍歷**/
    public void inList(TreeNode root)
    {
        if (root != null)
        {
            inList(root.getLeft());
            System.out.print(root.getData() + ",");
            inList(root.getRight());
        }
    }

    public TreeNode getRoot()
    {
        return root;
    }

    public void setRoot(TreeNode root)
    {
        this.root = root;
    }

    public int getSize()
    {
        return size;
    }

    public void setSize(int size)
    {
        this.size = size;
    }
    /**線索化二叉樹**/
    public void inThread(TreeNode root) {
        if ( root != null ) {
            // 線索化左孩子
            inThread(root.getLeft());
            // 左孩子為空
            if ( null == root.getLeft() )
            {
                // 將左孩子設(shè)置為線索
                root.setLeftIsThread(true);
                root.setLeft(pre);
            }
            // 右孩子為空
            if ( pre != null && null == pre.getRight() )
            {
                pre.setRightIsThread(true);
                pre.setRight(root);
            }
            pre = root;
            // 線索化右孩子
            inThread(root.getRight());
        }
    }
    /**
     * 中序遍歷線索二叉樹
     *
     */
    public void inThreadList(TreeNode root)
    {
        if (root != null)
        {
            // 如果左孩子不是線索
            while (root != null && !root.isLeftIsThread())
            {
                root = root.getLeft();
            }

            do
            {
                // 如果右孩子是線索
                System.out.print(root.getData() + ",");
                if (root.isRightIsThread())
                {
                    root = root.getRight();
                }
                // 有右孩子
                else
                {
                    root = root.getRight();
                    while (root != null && !root.isLeftIsThread())
                    {
                        root = root.getLeft();
                    }
                }
            } while (root != null);
        }
    }
}

package com.testtree;

/**
 * @author pier
 */
public class Test {
    public static void main(String[] args) {
    //不要問我為什么設(shè)置這么大，結(jié)尾看我效果截圖
        int[] arr = new int[10000];
        for (int i = 0; i < arr.length; i++) {
            arr[i]=i+1;
        }
        //創(chuàng)建一顆二叉樹
        BiTree biTree = new BiTree(arr);
        //中序遍歷二叉樹
        System.out.println("中序遍歷結(jié)果如下:");
        long start1 = System.currentTimeMillis();
        biTree.inList(biTree.getRoot());
        long end1 = System.currentTimeMillis();
        System.out.println();
        System.out.println("普通遍歷時(shí)間為:"+(end1-start1)+"毫秒");
        System.out.println("\n");
        //中序遍歷將二叉樹線索化
        biTree.inThread(biTree.getRoot());
        System.out.println("線索二叉樹中序遍歷如下:");
        long start2 = System.currentTimeMillis();
        biTree.inThreadList(biTree.getRoot());
        long end2 = System.currentTimeMillis();
        System.out.println();
        System.out.println("線索二叉樹的遍歷時(shí)間為:"+(end2-start2)+"毫秒");

    }
}