javascript - 在 es6 中查找我的树级宽度算法中的错误

Question

我正在练习解决算法问题，需要帮助找出我的代码中的错误。

--- 方向给定树的根节点，返回一个数组，其中每个元素都是宽度每个级别的树。

我试过在它上面运行测试用例，也在 JSBin 中运行它，但没有成功。运行测试时出现此错误:TypeError: undefined is not iterable

// my node class
class Node {
    constructor(data) {
    this.data = data;
    this.children = [];
    }

    add(data) {
        this.children.push(new Node(data));
    }
};

function levelWidth(root) {
    const store = [];
    const widths = [0];
    if(root){
        store.push(root);
        store.push('stop');
    }
    while(store.length > 1){
        const value = store.shift();
        if(value === 'stop'){
            widths.push(0);
            store.push(stop);
        }
        else {
            store.push(...value.children);
            widths[widths.length - 1]++;
        }
  }
  return widths;
}

运行时

    expect(levelWidth(root)).toEqual([1, 3, 2]);

我希望获得 [1,3,2] 的数组，但却得到 TypeError: undefined is not iterable for

    store.push(...value.children);

据我所知，我正确地使用了展开运算符？

Answer 1

您当前的错误是由于 undefined variable stop 引起的；即，您的表达式 store.push(stop); 引用了一个 undefined variable stop。

注释掉该行，特定错误不再是问题:

// my node class
class Node {
  constructor(data) {
    this.data = data;
    this.children = [];
  }

  add (data) {
    this.children.push(new Node(data));
  }
};

function levelWidth (root) {
  const store = [];
  const widths = [0];
  if (root) {
    store.push(root);
    store.push('stop');
  }
  while (store.length > 1) {
    const value = store.shift();
    if (value === 'stop') {
      widths.push(0);
      // store.push(stop);
    }
    else {
      store.push(...value.children);
      widths[widths.length - 1]++;
    }
  }
  return widths;
}