在嵌套树中定位元素 - JavaScript

时间:2016-12-16 23:22:12

标签: javascript dom search tree breadth-first-search

编写函数以搜索示例树中的元素时需要帮助。

var sampletree = [
  "a",
  "b",
  "c",
  [
    "d",
    "e",
    [
      "f",
      "h",
      "i",
      [
        "z",
        "x"
      ]
    ]
  ],
  [
    "y",
    "q",
    "t",
    [
      "m",
      "n",
      [
        "o",
        "p"
      ],
      [
        "r",
        "s",
        [
          "u",
          "v"
        ]
      ]
    ]
  ],
  "g"
]

希望通过仅使用javascript并且没有框架/库来搜索树中的o,p,u,v等内部元素来返回true / false。

我使用for循环来运行然后.indexOf但是无法获得它。

1 个答案:

答案 0 :(得分:2)

您可以使用此功能:

function findNodeBFS(tree, node) {
    var queue = tree.slice();
    for (var i = 0; i < queue.length; i++) {
        if (queue[i] === node) return true;
        if (Array.isArray(queue[i])) queue = queue.concat(queue[i]);
    }
    return false;
}

var sampletree = [ "a", "b", "c", [ "d", "e", [ "f", "h", "i", [ "z", "x" ] ] ], [ "y", "q", "t", [ "m", "n", [ "o", "p" ], [ "r", "s", [ "u", "v" ] ] ] ], "g" ];

console.log(findNodeBFS(sampletree, "u")); // true
console.log(findNodeBFS(sampletree, "j")); // false

解释

此算法维护 queue ,您可以将其视为一种“待办事项”列表。它从树的副本开始。代码遍历节点,但仅限于顶级节点。每当遇到一个代表更深层次的数组时,这些子代就被添加到队列的末尾,这意味着:“我稍后会与你打交道”

这是在这行代码中发生的事情:

if (Array.isArray(queue[i])) queue = queue.concat(queue[i]);

因此,如果queue[i]是一个数组,则其元素将附加 1 (浅拷贝)到队列中。

1 实际上,concat不会改变给定数组,而是生成一个新数组,这是队列和<的串联强>队列[I] 即可。通过将新阵列分配回队列,我们可以获得附加效果。取而代之的是我们可以完成 []。push.apply(queue,queue [i]),它通过追加第二个就地改变第一个。但它可能看起来更加神秘。

请注意,我们不附加我们找到的数组,但每个元素都在中。因此,当循环到达附加数据时,它实际上将访问原始树的更深层次。这确实是BFS的意思:首先访问当前级别上的节点,并且只在完成当前级别后处理更深层次的节点。队列(先进先出)是理想的数据结构。

深度优先搜索变体

DFS变体将是这种递归ES6函数:

function findNodeDFS(tree, node) {
    return Array.isArray(tree) && tree.some( val => findNode(val, node) ) 
           || node === tree;
}