[leetcode] 二叉樹的鋸齒形層序遍歷 (雙端隊列)

給你二叉樹的根節(jié)點 root ,返回其節(jié)點值的 鋸齒形層序遍歷 。(即先從左往右,再從右往左進行下一層遍歷,以此類推,層與層之間交替進行)。

示例:

輸入:root = [3,9,20,null,null,15,7]
輸出:[[3],[20,9],[15,7]]

自己的解法:利用stack和lStack兩個隊列,在lStack中改控制方向

function zigzagLevelOrder(root: TreeNode | null): number[][] {
  let res = []
  let rtl = true
  let stack: Array<TreeNode> = []
  if (root === null) return []
  stack.push(root)
  while (stack.length) {
    let size = stack.length
    let lStack = []
    res.push([])
    rtl = !rtl
    for(let i = 1; i <= size; i++) {
      let node = stack.shift()
      res[res.length - 1].push(node.val)
      lStack.push(node)
    }
    while(lStack.length) {
      let lNode = lStack.pop()
      if (rtl) {
        if (lNode.left) stack.push(lNode.left)
        if (lNode.right) stack.push(lNode.right)
      } else {
        if (lNode.right) stack.push(lNode.right)
        if (lNode.left) stack.push(lNode.left)
      }
    }
  }
  return res
};

其實利用雙端隊列更簡便,只需要一個隊列完成需求:

  • 如果從左至右,我們每次將被遍歷到的元素插入至雙端隊列的末尾。
  • 如果從右至左,我們每次將被遍歷到的元素插入至雙端隊列的頭部。
function zigzagLevelOrder(root: TreeNode | null): number[][] {
    if (!root) {
        return [];
    }

    const ans = [];
    const nodeQueue = [root];
    let isOrderLeft = true;

    while (nodeQueue.length) {
        let levelList = [];
        const size = nodeQueue.length;
        for (let i = 0; i < size; ++i) {
            const node = nodeQueue.shift();
            if (isOrderLeft) {
                levelList.push(node.val);
            } else {
                levelList.unshift(node.val);
            }
            if (node.left !== null) {
                nodeQueue.push(node.left);
            }
            if (node.right !== null) {
                nodeQueue.push(node.right);
            }
        }            
        ans.push(levelList);
        isOrderLeft = !isOrderLeft;
    }
    return ans;
};
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容