Question

我尝试从给定的级别订单（BFS订单）构建BST。我知道这是可能的，但我不知道怎么写这个。问题是，我必须使用BFS序列。所以，我不能在这里使用递归，我不得不写一下我的程序...而且我觉得有点混乱。

我试着这样做：

public static TreeNode constructBFSTree(ArrayList<Integer> bfs) {
    if (bfs== null) return null;
    ArrayList<TreeNode> result = new ArrayList<TreeNode>();
        for (int i = 0; i < bfs.size()-1; i ++){
             TreeNode node = result.get(i);
             int leftValue = (bfs.get(i+1)!=null)? bfs.get(i+1) : Integer.MAX_VALUE ;
             int rightValue = (bfs.get(i+2)!=null)? bfs.get(i+2) : Integer.MIN_VALUE;

             node.left = (leftValue <= node.data)? new TreeNode(leftValue) : null;
             node.right = (rightValue > node.data)? new TreeNode(rightValue) : null;
             result.add(node);
        }

        return result.get(0);
    }

本地ArrayList在这里并不重要。我只是添加它来“捕获”第一个节点，它是我应该返回的构造树的根。问题是我只得到了根和它的孩子。

我该如何编写这个程序？

Answer 1

您如何尝试以下代码？（注意：我还没有测试过它，因为你还没有提供类定义。但是它应该会把你推向正确的方向。）

我对TreeNode类的假设是它的构造函数采用整数，并且它将left和right指针初始化为null。例如：

class TreeNode {
    TreeNode left;
    TreeNode right;
    int key;
    public TreeNode(int key) {
        this.key = key;
        this.left = null;
        this.right = null;
    }
}

该函数的代码可以如下：

public static TreeNode constructBFSTree(ArrayList<Integer> bfs) {
    if (bfs == null || bfs.isEmpty()) {
        return null;
    }

    Queue<TreeNode> q = new Queue<TreeNode>();
    TreeNode root = new TreeNode(bfs.get(0));
    q.add(root);
    int i = 1;
    while (!q.isEmpty() && i < bfs.size()) {
        TreeNode currentNode = q.poll();
        currentNode.left = new TreeNode(bfs.get(i++));
        q.add(curentNode.left);
        if (i < bfs.length()) {
            currentNode.right = new TreeNode(bfs.get(i++));
            q.add(currentNode.right);
        }
    }
    return root;
}

从级别订单构建BST（BFS订单）

1 个答案: