堆栈+优化 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

解题
堆栈：
遍历到节点——放入；
遍历到 # + 栈顶为 # ——节点出栈，放入#；

class Solution {
public:
    bool isValidSerialization(string preorder) {
        //堆栈储存，连续两个#一个顶点出堆
        //若堆空还未遍历完——false
        stack<string> node;
        i=0;
        while(i<preorder.length()){  //遍历完成
            string tmp="";
            if(isdigit(preorder[i])) tmp=getnum(preorder);
            else if(preorder[i]=='#') tmp="#";
        
            while(!node.empty()&&node.top()=="#"&&tmp=="#"){
                node.pop();
                if(node.empty()) return false;
                node.pop();
            }
            
            if(tmp!="")node.push(tmp);
            i++;
        }
        return node.size()==1&&node.top()=="#";
    }
private:
    int i;
    string getnum(string s){
        string res="";
        while(i<s.length()&&isdigit(s[i])){
            res+=s[i++];
        }
        return res;
    }
};

优化
将节点栈用slot代替；
一开始空树，有一个节点位；
遇到数字，加一个节点位；
遇到#，减一个节点位；
若遍历过程节点位<0，则返回false；
判断遍历完成后树是否被填满——slot==0；

class Solution {
public:
    bool isValidSerialization(string preorder) {
        int i=0;
        int slot=1;
        while(i<preorder.length()){
            if(slot<=0) return false;
            if(isdigit(preorder[i])){
                while(i<preorder.length()&&isdigit(preorder[i])) i++; //到，
                slot+=1;  //可放#的位置+1
            }
            else if(preorder[i]=='#')
                slot-=1;
            i++;
        }
        return slot==0;
    }
};