두 갈래 트리 - 최소 경로

최소 경로


두 갈래 나무의 최소 경로는 두 갈래 나무의 뿌리 노드가 잎 노드에 도달하는 경로 중 가장 짧은 거리를 가리킨다.
//recursively
public int minDepth1(TreeNode root) {
    if (root == null) {
        return 0;
    } else if (root.left != null && root.right != null) {
        return 1+Math.min(minDepth(root.left), minDepth(root.right));
    } else {
        return 1+Math.max(minDepth(root.left), minDepth(root.right));
    }
}

// iteratively, BFS
public int minDepth(TreeNode root) {
    if (root == null) {
        return 0;
    }
    Queue<TreeNode> queue = new LinkedList<>();
    queue.add(root);
    int depth = 1;
    while (!queue.isEmpty()) {
        int l = queue.size();
        for (int i = 0; i < l; i++) {
            TreeNode n = queue.poll();
            if (n.left == null && n.right == null) {
                return depth;
            } 
            if (n.left != null) {
                queue.add(n.left);
            }
            if (n.right != null) {
                queue.add(n.right);
            }
        }
        depth++;
    }
    return depth;
}

좋은 웹페이지 즐겨찾기