[路飞] 31——leetcode - [331] 验证二叉树的前序序列化

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

331. 验证二叉树的前序序列化

题目分析

序列化二叉树的一种方法是使用 前序遍历 。当我们遇到一个非空节点时，我们可以记录下这个节点的值。如果它是一个空节点，我们可以使用一个标记值记录，例如 #。

例如，上面的二叉树可以被序列化为字符串 "9,3,4,#,#,1,#,#,2,#,6,#,#"，其中 # 代表一个空节点。

给定一串以逗号分隔的序列，验证它是否是正确的二叉树的前序序列化。编写一个在不重构树的条件下的可行算法。

• 保证 每个以逗号分隔的字符或为一个整数或为一个表示 null 指针的 '#' 。
• 可以认为输入格式总是有效的，例如它永远不会包含两个连续的逗号，比如 "1,,3" 。

注意：不允许重建树。

思路讲解

前序遍历：中前后

示例

输入: preorder = "9,3,4,#,#,1,#,#,2,#,6,#,#"
输出: true

输入: preorder = "1,#"
输出: false
 
输入: preorder = "9,#,#,1"
输出: false

代码

/*
 * @lc app=leetcode.cn id=331 lang=javascript
 *
 * [331] 验证二叉树的前序序列化
 */

// @lc code=start
/**
 * @param {string} preorder
 * @return {boolean}
 */
var isValidSerialization = function(preorder) {
  let n = preorder.length
  let stack = [1]
  let i = 0
  while (i < n) {
    if (!stack.length) return false
    if (preorder[i] === ',') i++
    else if (preorder[i] === '#') {
      // 只减不增
      stack[stack.length - 1]--
      if (stack[stack.length - 1] === 0) {
        stack.pop()
      }
      i++
    } else {
      while (i < n && preorder[i] !== ',') {
        i++
      }
      stack[stack.length - 1]--
      if (stack[stack.length - 1] == 0) {
        stack.pop()
      }
      stack.push(2)
    }
  }
  return !stack.length
};
// @lc code=end