597. 최대 평균수를 가진 자수

1797 단어
묘사
두 갈래 나무 한 그루를 주어 최대 평균치가 있는 나무를 찾아라.하위 트리의 루트 결점을 되돌려줍니다.
주의사항
LintCode는 루트 결점을 출력하여 노드로 돌아가는 하위 트리를 출력합니다. 최대 평균 하위 트리는 한 그루뿐이라고 보장합니다.
예제
두 갈래 나무:
     1
   /   \
 -5     11
 / \   /  \
1   2 4    -2 

노드 11을 반환합니다.

코드

Traverse + Divide Conquer  
 /**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */
public class Solution {
    private class ResultType {
        public int size;
        public int sum;
        public ResultType (int sum, int size) {
            this.sum = sum;
            this.size = size;
        }
    }
    
    // subtree 
    // subtreeResult ResultType
    private TreeNode subtree = null;
    private ResultType subtreeResult = null;
    
    public TreeNode findSubtree2(TreeNode root) {
        helper(root);
        return subtree;
    }
    
    private ResultType helper(TreeNode root) {
        if (root == null) {
            return new ResultType(0, 0);
        }
        
        ResultType left = helper(root.left);
        ResultType right = helper(root.right);
        // result 
        ResultType result = new ResultType(
            left.sum + right.sum + root.val,
            left.size + right.size + 1
        );
      
        /*  subtreeResult.sum / subtreeResult.size < 
         * result.sum / result.size 
         */
        // result , result
        if (subtree == null || 
            result.sum * subtreeResult.size > subtreeResult.sum * result.size) {
            subtree = root;
            subtreeResult = result;
        }
        return result;
    } 
}

좋은 웹페이지 즐겨찾기