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;
}
};