|
|
package class_2022_12_1_week;
|
|
|
|
|
|
// 给你一棵 二叉树 的根节点 root ，树中有 n 个节点
|
|
|
// 每个节点都可以被分配一个从 1 到 n 且互不相同的值
|
|
|
// 另给你一个长度为 m 的数组 queries
|
|
|
// 你必须在树上执行 m 个 独立 的查询，其中第 i 个查询你需要执行以下操作：
|
|
|
// 从树中 移除 以 queries[i] 的值作为根节点的子树
|
|
|
// 题目所用测试用例保证 queries[i] 不 等于根节点的值。
|
|
|
// 返回一个长度为 m 的数组 answer ，其中 answer[i] 是执行第 i 个查询后树的高度。
|
|
|
// 注意：
|
|
|
// 查询之间是独立的，所以在每个查询执行后，树会回到其 初始 状态。
|
|
|
// 树的高度是从根到树中某个节点的 最长简单路径中的边数 。
|
|
|
// 测试链接 : https://leetcode.cn/problems/height-of-binary-tree-after-subtree-removal-queries/
|
|
|
public class Code02_HeightAfterSubtreeRemoval {
|
|
|
|
|
|
// 提交时不用提交这个类
|
|
|
public static class TreeNode {
|
|
|
public int val;
|
|
|
public TreeNode left;
|
|
|
public TreeNode right;
|
|
|
}
|
|
|
|
|
|
// 提交如下方法
|
|
|
public static final int MAXN = 100010;
|
|
|
public static int[] dfn = new int[MAXN];
|
|
|
public static int[] deep = new int[MAXN];
|
|
|
public static int[] size = new int[MAXN];
|
|
|
public static int[] maxl = new int[MAXN];
|
|
|
public static int[] maxr = new int[MAXN];
|
|
|
public static int n;
|
|
|
|
|
|
public static int[] treeQueries(TreeNode root, int[] queries) {
|
|
|
n = 0;
|
|
|
// 每个val，编号
|
|
|
// 每个val，深度
|
|
|
// 每个val的子树，大小
|
|
|
dfs(root, 0);
|
|
|
for (int i = 1; i <= n; i++) {
|
|
|
maxl[i] = Math.max(maxl[i - 1], deep[i]);
|
|
|
}
|
|
|
maxr[n + 1] = 0;
|
|
|
for (int i = n; i >= 1; i--) {
|
|
|
maxr[i] = Math.max(maxr[i + 1], deep[i]);
|
|
|
}
|
|
|
int m = queries.length;
|
|
|
int[] ans = new int[m];
|
|
|
for (int i = 0; i < m; i++) {
|
|
|
// queries[i] -> a
|
|
|
// a -> 编号x
|
|
|
// a -> 子树大小
|
|
|
// x ... 子树大小这么多范围 删掉
|
|
|
int leftMax = maxl[dfn[queries[i]] - 1];
|
|
|
int rightMax = maxr[dfn[queries[i]] + size[dfn[queries[i]]]];
|
|
|
ans[i] = Math.max(leftMax, rightMax);
|
|
|
}
|
|
|
return ans;
|
|
|
}
|
|
|
|
|
|
// n = 0 1 2 3 4 5 6 7
|
|
|
public static void dfs(TreeNode head, int h) {
|
|
|
int i = ++n;
|
|
|
dfn[head.val] = i;
|
|
|
deep[i] = h;
|
|
|
size[i] = 1;
|
|
|
if (head.left != null) {
|
|
|
dfs(head.left, h + 1);
|
|
|
size[i] += size[dfn[head.left.val]];
|
|
|
}
|
|
|
if (head.right != null) {
|
|
|
dfs(head.right, h + 1);
|
|
|
size[i] += size[dfn[head.right.val]];
|
|
|
}
|
|
|
}
|
|
|
|
|
|
}
|
