type TreeNode {children?: TreeNode[][key: string]: any}/*** 给树结构补充 canSelect 字段* 规则* 1. 当前级别 3可选* 2. 当前级别 3但没有子节点也可选* 3. 其他不可选** param tree 树数据* param level 起始层级默认从 1 开始* returns 新树*/export function setTreeCanSelectT extends TreeNode(tree: T[],level 1): ArrayT { canSelect: boolean } {if (!Array.isArray(tree)) return []return tree.map((node) {const children Array.isArray(node.children) ? node.children : []const hasChildren children.length 0const canSelect level 3 || !hasChildrenreturn {...node,canSelect,children: hasChildren? setTreeCanSelect(children, level 1): []}})}取值type TreeNode {label: stringvalue: string | numberchildren?: TreeNode[][key: string]: any}/*** 树拍平*/function flattenTree(tree: TreeNode[]): TreeNode[] {const result: TreeNode[] []const dfs (nodes: TreeNode[]) {nodes.forEach((node) {result.push(node)if (node.children node.children.length) {dfs(node.children)}})}dfs(tree)return result}/*** 根据 valueList 获取节点对象列表*/export function getNodesByValues(tree: TreeNode[],valueList: Arraystring | number): TreeNode[] {if (!Array.isArray(tree) || !Array.isArray(valueList)) return []const flatList flattenTree(tree)const valueSet new Set(valueList)return flatList.filter((node) valueSet.has(node.value))}