LeetCode 331. 验证二叉树的前序序列化--二叉树+栈

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

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

 _9_
/   \

3 2
/ \ /
4 1 # 6
/ \ / \ / \

# # # #

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

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

每个以逗号分隔的字符或为一个整数或为一个表示 null 指针的 ‘#’ 。

你可以认为输入格式总是有效的，例如它永远不会包含两个连续的逗号，比如 “1,3” 。

示例 1:

输入: “9,3,4,#,#,1,#,#,2,#,6,#,#”
输出: true

示例 2:

输入: “1,#”
输出: false

示例 3:

输入: “9,#,#,1”
输出: false

题解

思路比较简单，因为题目说明的也比较详细，也就是每个叶子节点尾部肯定有两个#，于是我们通过这两个#不断收敛，那么肯定最后只会剩一个字符#，所以我们利用栈进行判断比较方便。

AC代码

class Solution {
public:
    vector<string>q;
    stack<string>q2;
    bool isValidSerialization(string preorder) {
        string t="";
        for(int i=0;i<preorder.length();i++)
        {
            if(preorder[i]==',')
            {
                q.push_back(t);
                t="";
            }
            else t+=preorder[i];
        }
        q.push_back(t);
        if(q.size()==1&&q[0]=="#")return true;
        if(q.size()<3)return false;
        for(int i=0;i<q.size();i++)
        {
            q2.push(q[i]);
            while(q2.size()>=3&&q2.top()=="#")
            {
                q2.pop();
                if(q2.top()=="#")
                {
                    q2.pop();
                    if(q2.top()=="#")
                    {
                        return false;
                    }
                    q2.pop();
                    q2.push("#");
                }
                else
                {
                    q2.push("#");
                    break;
                }
            }
        }
        return q2.size()==1?true:false;
    }
};