移动 N 叉树的子树

// 题目思路：
// 给定一个 N 叉树，和两个节点 `node` 和 `newParent`。
// 要求把 `node` 及其子树移动到 `newParent` 的子节点中。
// 假设输入的树节点具有以下结构：
// class Node {
//     public int val;
//     public List<Node> children;
//     public Node() {}
//     public Node(int _val) {
//         val = _val;
//     }
//     public Node(int _val, List<Node> _children) {
//         val = _val;
//         children = _children;
//     }
// }

import java.util.*;
import java.util.stream.*;

class Solution {
    public Node moveSubTree(Node root, Node node, Node newParent) {
        if (root == null || node == null || newParent == null) return root;

        // 如果 newParent 是 node 或 node 的子节点，直接返回 root
        if (isDescendant(node, newParent)) return root;

        // 移除 node
        removeNode(root, node);

        // 将 node 添加到 newParent 的子节点
        newParent.children = Stream.concat(newParent.children.stream(), Stream.of(node))
                                    .collect(Collectors.toList());

        return root;
    }

    private boolean isDescendant(Node node, Node target) {
        return node == target || node.children.stream().anyMatch(child -> isDescendant(child, target));
    }

    private void removeNode(Node parent, Node node) {
        parent.children = parent.children.stream()
                              .filter(child -> !child.equals(node))
                              .collect(Collectors.toList());
        parent.children.forEach(child -> removeNode(child, node));
    }
}