jd1712.BiNode

题目描述

二叉树数据结构TreeNode可用来表示单向链表（其中left置空，right为下一个链表节点）。实现一个方法，把二叉搜索树转换为单向链表，要求依然符合二叉搜索树的性质，转换操作应是原址的，也就是在原始的二叉搜索树上直接修改。

返回转换后的单向链表的头节点。

注意：本题相对原题稍作改动

示例:

输入： [4,2,5,1,3,null,6,0]
输出： [0,null,1,null,2,null,3,null,4,null,5,null,6]

题解

中序遍历迭代

因为要符合二叉搜索树的性质, 所以要保持顺序, 因此采用中序遍历的方法.

每遍历到一个节点, 需要将当前的左节点置空, 然后接在上一个节点上, 作为上一个节点的右节点. 因为要时刻记录上一个节点的情况, 所以提前初始化一个哨兵节点, 随着遍历的进行, 这个哨兵节点也不断移动.

所以算法大致流程为, 在中序遍历的基础上:

1. 将当前从栈中弹出的节点左子节点置空
2. 将当前从栈中弹出的节点接在上一个节点的右边
3. 将上一个节点移动到当前节点处

public TreeNode convertBiNode(TreeNode root) {
        TreeNode head = new TreeNode(0);
        TreeNode prev = head;

        Stack<TreeNode> stack = new Stack<>();
        TreeNode cur = root;

        while (cur != null || !stack.isEmpty()) {
            if (cur!=null){
                stack.push(cur);
                cur = cur.left;
            }else {
                cur = stack.pop();
                cur.left = null;
                prev.right = cur;
                prev = cur;
                cur = cur.right;
            }
        }

        return head.right;
    }

---

中序遍历递归

思路跟迭代法一样, 都是在遍历到当前节点时进行操作

public TreeNode convertBiNode(TreeNode root) {
    TreeNode head = new TreeNode(0);
    inorder(root, head);
    return head.right;
}

private TreeNode inorder(TreeNode root, TreeNode prev) {
    if (root!= null){
        prev = inorder(root.left, prev);
        root.left = null;
        prev.right = root;
        prev = root;
        prev = inorder(root.right, prev);
    }
    return prev;
}