找到所有非罗马的节点，其所有后代都是二叉树中的罗马

Question

问题：
设T是二叉树。将罗马节点定义为T中的节点v，使得v的左子树中的后代数与v右子树中的节点数相差最多5.描述用于查找T的每个节点v的线性时间算法，这样v不是罗马节点，但所有v的后代都是罗马节点。

到目前为止我有什么：
我可以想到O(n^2)（自上而下的方法）解决方案，我将遍历树并检查节点是否不是罗马，然后遍历此节点的后代以检查它们是否都是罗马的。所以，通过这种方式，我每次遍历每个节点 我假设有一种自下而上的方法，可以在O(n)中找到所需的节点。

有什么想法吗？

Answer 1

谢谢@n.m.对你的评论。我已经使用您的输入实现了解决方案。
的逻辑：
基本逻辑是将节点的isRoman字段设置为false，即使节点的后代（假设盛大的孩子）不是罗马字符。这意味着即使节点满足属性（左右后代最多为5），但如果其左右子树isRoman为false，则当前节点为isRoman也将是false。

/*
 * public class TreeNode {
 * int val;
 * int size;        //stores the number of descendants including itself
 * boolean isRoman; // is true when this node is roman and all its descendants are roman, 
 *                  //false even when a grand grand child node of this node is not roman.
 * TreeNode left;
 * TreeNode right;
 * }
 */
public static void roman(TreeNode root, List<TreeNode> lst){
    if(root == null)
        return;
    if(root.left == null && root.right == null){
        root.size = 1;
        root.isRoman = true;
        return;
    }
    int left = 0;
    int right = 0;
    boolean isLeftRoman = false;
    boolean isRightRoman = false;

    if(root.left != null) {
        roman(root.left,lst);
        left = root.left.size;
        isLeftRoman = root.left.isRoman;
    }else{
     isLeftRoman=true;
    }


    if(root.right != null) {
        roman(root.right,lst);
        right = root.right.size;
        isRightRoman = root.right.isRoman;
    }else{
  isRightRoman=true;
        }

    //if the current node is roman and all it's descendants are roman, update isRoman to true
    if(Math.abs(left-right) <= 5 && isLeftRoman && isRightRoman)
        root.isRoman = true;
    else
        root.isRoman = false;

    //update the current node's size
    root.size = left+right+1;

    //add the node to list which is not roman but all its descendants are roman
    if(Math.abs(left-right) > 5 && isLeftRoman && isRightRoman)
        lst.add(root); 
}