在Java中运行一棵树

Question

基本上我想写一个函数，给定一个决策树，打印树中所有可能的路径，并且对于每个可能的路径，还打印一个布尔值列表（真或假），指示它是否离开(true) 或 right(false) 在树中。

这是我的树类型：

public class Tree {

private TreeNode root;


private static class TreeNode {
    private Pair<String, Float> data;
    private TreeNode leftNode;
    private TreeNode rightNode;

    private TreeNode( Pair<String, Float> data, TreeNode left, TreeNode right) {
        this.data = data;
        this.leftNode = left;
        this. rightNode = right;
        }
    }
}

这是我到目前为止所做的事情：

private void getTreePaths(TreeNode r, List<Pair<String, Float>> list,
        List<String> flags) {

    if (r == null)
        return;

    // adds the node to the list
    list.add(r.data);

    // goes left
    if (r.leftNode != null) {

        //went left
        flags.add("true");
        getTreePaths(r.leftNode, new ArrayList<Pair<String, Float>>(list), flags);
    }

    //instead goes right
    else if (r.rightNode != null) {
        //went right
        flags.add("false");
        getTreePaths(r.rightNode, new ArrayList<Pair<String, Float>>(list), flags);
    }

    else if (r.leftNode == null && r.rightNode == null) {
        System.out.println(list.toString());
        System.out.println(flags.toString());
        return;
    }
}

我的问题是这只会打印第一条路径。

我该如何解决这个问题？

Answer 1

什么是getTreeRules？那应该是getTreePaths吗？您需要为每个节点回调函数。

在评论中编辑您的最后一个问题：

您的标志列表正在累积，因为您要向左添加，然后在向右移动时保留它。您可以尝试以下方法：

private void getTreePaths(TreeNode r, List<Pair<String, Float>> list,
        List<String> flags, boolean lastmove) {

    flags.add(lastmove);

    if (r == null)
        return;

    // adds the node to the list
    list.add(r.data);

    // goes left
    if (r.leftNode != null) {

        //went left
        getTreePaths(r.leftNode, new ArrayList<Pair<String, Float>>(list), flags, true);
    }

    //go right
    if (r.rightNode != null) {
        //went right
        getTreePaths(r.rightNode, new ArrayList<Pair<String, Float>>(list), flags, false);
    }

    if (r.leftNode == null && r.rightNode == null) {
        System.out.println(list.toString());
        flags.remove(0); //remove first element because you passed in false value with root
        System.out.println(flags.toString());
        return;
    }
}